google-deepmind / mctx
Monte Carlo tree search in JAX
☆2,357Updated 3 months ago
Related projects ⓘ
Alternatives and complementary repositories for mctx
- ☆2,688Updated 6 months ago
- ☆845Updated 4 months ago
- ☆692Updated last year
- ☆1,263Updated last month
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,223Updated last year
- Really Fast End-to-End Jax RL Implementations☆724Updated 2 months ago
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆804Updated last month
- JAX-based neural network library☆2,909Updated last week
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆638Updated this week
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆870Updated 11 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,679Updated this week
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.☆1,016Updated this week
- Optax is a gradient processing and optimization library for JAX.☆1,699Updated last week
- High throughput synchronous and asynchronous reinforcement learning☆829Updated this week
- ♟️ Vectorized RL game environments in JAX☆412Updated this week
- A Production-ready Reinforcement Learning AI Agent Library brought by the Applied Reinforcement Learning team at Meta.☆2,672Updated this week
- Massively parallel rigidbody physics simulation on accelerator hardware.☆2,348Updated last week
- A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.☆2,346Updated this week
- A library of reinforcement learning components and agents☆3,517Updated 3 weeks ago
- A suite of test scenarios for multi-agent reinforcement learning.☆621Updated 2 weeks ago
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.☆592Updated this week
- ☆508Updated 9 months ago
- RL Environments in JAX 🌍☆652Updated 4 months ago
- A platform for managing machine learning experiments☆818Updated 3 months ago
- Cramming the training of a (BERT-type) language model into limited compute.☆1,296Updated 5 months ago
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,481Updated this week
- Mastering Diverse Domains through World Models☆1,381Updated 3 months ago
- Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research☆708Updated 2 weeks ago
- ☆751Updated this week
- Mastering Atari with Discrete World Models☆900Updated last year