google-deepmind / mctxLinks
Monte Carlo tree search in JAX
☆2,494Updated last month
Alternatives and similar repositories for mctx
Users that are interested in mctx are comparing it to the libraries listed below
Sorting:
- ☆898Updated 11 months ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆901Updated last year
- JAX-based neural network library☆3,041Updated last week
- Really Fast End-to-End Jax RL Implementations☆882Updated 8 months ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,381Updated 2 weeks ago
- ☆1,320Updated 3 weeks ago
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,562Updated last month
- ♟️ Vectorized RL game environments in JAX☆482Updated 3 months ago
- Optax is a gradient processing and optimization library for JAX.☆1,908Updated this week
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.☆1,062Updated 3 weeks ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆731Updated last month
- ☆709Updated last year
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆838Updated 7 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,784Updated last month
- ☆2,754Updated last year
- A library of reinforcement learning components and agents☆3,697Updated last week
- Massively parallel rigidbody physics simulation on accelerator hardware.☆2,708Updated last week
- The NetHack Learning Environment☆953Updated last year
- A Production-ready Reinforcement Learning AI Agent Library brought by the Applied Reinforcement Learning team at Meta.☆2,838Updated last week
- RL Environments in JAX 🌍☆763Updated last week
- Flax is a neural network library for JAX that is designed for flexibility.☆6,586Updated last week
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,265Updated 5 months ago
- ☆856Updated this week
- A platform for managing machine learning experiments☆857Updated 2 weeks ago
- MuZero☆2,652Updated 9 months ago
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.☆774Updated this week
- A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.☆2,804Updated this week
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,422Updated last month
- maximal update parametrization (µP)☆1,533Updated 10 months ago
- A suite of test scenarios for multi-agent reinforcement learning.☆705Updated 2 weeks ago