google-deepmind / mctxLinks
Monte Carlo tree search in JAX
☆2,534Updated 3 weeks 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☆949Updated last year
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆911Updated last year
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆841Updated 11 months ago
- ☆2,788Updated last year
- Code for Cicero, an AI agent that plays the game of Diplomacy with open-domain natural language negotiation.☆1,384Updated 5 months ago
- ♟️ Vectorized RL game environments in JAX☆524Updated 6 months ago
- ☆718Updated 2 years ago
- ☆1,355Updated 3 weeks ago
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,284Updated 9 months ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆756Updated 3 months ago
- ☆915Updated last year
- A Production-ready Reinforcement Learning AI Agent Library brought by the Applied Reinforcement Learning team at Meta.☆2,931Updated last week
- High throughput synchronous and asynchronous reinforcement learning☆938Updated 3 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,820Updated 3 months ago
- Massively parallel rigidbody physics simulation on accelerator hardware.☆2,864Updated this week
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,602Updated this week
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.☆1,087Updated this week
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools, with 10x faster training through evolutionary h…☆822Updated this week
- Mastering Diverse Domains through World Models☆2,141Updated 5 months ago
- A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.☆3,054Updated this week
- JAX-based neural network library☆3,093Updated this week
- A library of reinforcement learning components and agents☆3,780Updated 2 months ago
- RL Environments in JAX 🌍☆804Updated 3 months ago
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆623Updated last month
- Optax is a gradient processing and optimization library for JAX.☆2,017Updated last week
- Cramming the training of a (BERT-type) language model into limited compute.☆1,349Updated last year
- ☆498Updated 3 years ago
- An algorithm that generalizes the paradigm of self-play reinforcement learning and search to imperfect-information games.☆682Updated last year
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆844Updated last year
- Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research☆745Updated 3 weeks ago