google-deepmind / mctxLinks
Monte Carlo tree search in JAX
☆2,584Updated 4 months ago
Alternatives and similar repositories for mctx
Users that are interested in mctx are comparing it to the libraries listed below
Sorting:
- Really Fast End-to-End Jax RL Implementations☆1,013Updated last year
- Code for Cicero, an AI agent that plays the game of Diplomacy with open-domain natural language negotiation.☆1,405Updated 9 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,855Updated 7 months ago
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,623Updated 2 weeks ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆920Updated 2 years ago
- ♟️ Vectorized RL game environments in JAX☆579Updated 10 months ago
- ☆935Updated last year
- ☆1,402Updated last month
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.☆1,115Updated last week
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆800Updated last month
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆863Updated last year
- RL Environments in JAX 🌍☆852Updated 8 months ago
- JAX-based neural network library☆3,178Updated last week
- ☆728Updated 2 years ago
- High throughput synchronous and asynchronous reinforcement learning☆971Updated 2 months ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,748Updated this week
- ☆2,804Updated last year
- Optax is a gradient processing and optimization library for JAX.☆2,169Updated this week
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,299Updated last year
- A Production-ready Reinforcement Learning AI Agent Library brought by the Applied Reinforcement Learning team at Meta.☆2,971Updated 2 weeks ago
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools, with 10x faster training through evolutionary h…☆866Updated last week
- Massively parallel rigidbody physics simulation on accelerator hardware.☆3,034Updated 2 weeks ago
- Cramming the training of a (BERT-type) language model into limited compute.☆1,361Updated last year
- MuZero☆2,756Updated last year
- JAX - A curated list of resources https://github.com/google/jax☆2,026Updated last week
- Evolution Through Large Models☆737Updated 2 years ago
- Evolution Strategies in JAX 🦎☆719Updated 4 months ago
- Model API for GALACTICA☆2,739Updated 2 years ago
- A suite of test scenarios for multi-agent reinforcement learning.☆781Updated 3 weeks ago
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆668Updated 5 months ago