google-deepmind / mctx
Monte Carlo tree search in JAX
☆2,446Updated 3 months ago
Alternatives and similar repositories for mctx:
Users that are interested in mctx are comparing it to the libraries listed below
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆886Updated last year
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆832Updated 5 months ago
- ☆2,729Updated 11 months ago
- Really Fast End-to-End Jax RL Implementations☆840Updated 6 months ago
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.☆1,045Updated 3 weeks ago
- ☆1,300Updated 2 months ago
- ♟️ Vectorized RL game environments in JAX☆456Updated 3 weeks ago
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,541Updated last week
- Optax is a gradient processing and optimization library for JAX.☆1,839Updated this week
- JAX-based neural network library☆2,996Updated 2 weeks ago
- ☆890Updated 9 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,748Updated 3 months ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆709Updated this week
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,285Updated this week
- ☆707Updated last year
- MuZero☆2,604Updated 6 months ago
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,252Updated 3 months ago
- Massively parallel rigidbody physics simulation on accelerator hardware.☆2,597Updated last month
- High throughput synchronous and asynchronous reinforcement learning☆885Updated 3 months ago
- Llama 2 Everywhere (L2E)☆1,516Updated 2 months ago
- Flax is a neural network library for JAX that is designed for flexibility.☆6,443Updated this week
- ☆841Updated 2 weeks ago
- Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.☆2,518Updated 11 months ago
- RL Environments in JAX 🌍☆727Updated 8 months ago
- A platform for managing machine learning experiments☆841Updated last week
- Mastering Diverse Domains through World Models☆1,561Updated last month
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,610Updated last year
- tree is a library for working with nested data structures☆975Updated 2 months ago
- JAX - A curated list of resources https://github.com/google/jax☆1,747Updated last month
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.☆711Updated this week