mctx
Search algorithm library
An open-source library providing efficient implementations of search algorithms for reinforcement learning
Monte Carlo tree search in JAX
2k stars
28 watching
191 forks
Language: Python
last commit: 4 months ago
Linked from 1 awesome list
jaxmonte-carlo-tree-searchreinforcement-learning
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 jnptr/subgoal-mcts | An extension to Monte-Carlo Tree Search (MCTS) that efficiently generates macro-actions from subgoal predicates in deterministic domains. | 4 |
 matthias-wright/flaxmodels | Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 238 |
 eleutherai/gpt-neox | Provides a framework for training large-scale language models on GPUs with advanced features and optimizations. | 6,941 |
| google-deepmind/kfac-jax | Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks. | 248 |
 google/jaxopt | An open-source project providing hardware accelerated, batchable and differentiable optimizers in JAX for deep learning. | 933 |
 rlouf/mcx | Tools and methods for Bayesian deep learning using probabilistic programming. | 325 |
 resibots/mcts | An implementation of the Monte Carlo Tree Search algorithm in C++14 | 39 |
 deepset-ai/haystack | An AI orchestration framework to build customizable LLM applications with advanced retrieval methods. | 17,691 |
| google-deepmind/optax | 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. | 1,697 |
| google-deepmind/deepmind-research | Provides implementations and illustrative code to accompany DeepMind research publications | 13,250 |
 mjlaine/mcmcstat | A collection of Matlab functions for Bayesian analysis using Markov chain Monte Carlo simulation. | 86 |
| google/paxml | A framework for configuring and running machine learning experiments on top of Jax. | 457 |
| google/model_search | An automated framework for discovering optimal neural network architectures for machine learning tasks. | 3,268 |
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
| google-deepmind/dm_pix | An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research. | 389 |