mctx
Monte Carlo tree search in JAX
AI summary
Search algorithm library
An open-source library providing efficient implementations of search algorithms for reinforcement learning
- stars
- 2.4K
- forks
- 193
- watching
- 28
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
Subgoal-based MCTS
An extension to Monte-Carlo Tree Search (MCTS) that efficiently generates macro-actions from subgoal predicates in deterministic domains.
Model repository
Provides pre-trained deep learning models for the Jax/Flax ecosystem.
Language model trainer
Provides a framework for training large-scale language models on GPUs with advanced features and optimizations.
KFAC optimizer
Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks.
Deep learning optimizer
An open-source project providing hardware accelerated, batchable and differentiable optimizers in JAX for deep learning.
rlouf/mcx328
Probabilistic programming toolset
Tools and methods for Bayesian deep learning using probabilistic programming.
MCTS algo
An implementation of the Monte Carlo Tree Search algorithm in C++14
deepset-ai/haystack18.1K
LLM orchestrator
An AI orchestration framework to build customizable LLM applications with advanced retrieval methods.
Gradient optimizer library
A gradient processing and optimization library designed to facilitate research and productivity in machine learning by providing building blocks for custom optimizers and gradient processing components.
Research codebase
Provides implementations and illustrative code to accompany DeepMind research publications
MCMC simulator
A Matlab toolbox for Bayesian analysis using Markov chain Monte Carlo simulation
google/paxml461
ML framework
A framework for configuring and running machine learning experiments on top of Jax.
Architectural finder
An automated framework for discovering optimal neural network architectures for machine learning tasks.
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
Image processor
An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research.
Featured in 1 awesome list
Each link jumps to the spot where the list mentions mctx.
