google-deepmind / mctx
Monte Carlo tree search in JAX
☆2,350Updated 3 months ago
Related projects ⓘ
Alternatives and complementary repositories for mctx
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆872Updated 10 months ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,103Updated last week
- ☆1,258Updated last month
- JAX-based neural network library☆2,894Updated last week
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆800Updated 3 weeks ago
- Really Fast End-to-End Jax RL Implementations☆712Updated 2 months ago
- ☆839Updated 4 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,223Updated last year
- Optax is a gradient processing and optimization library for JAX.☆1,687Updated this week
- A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.☆2,314Updated this week
- A library of reinforcement learning components and agents☆3,503Updated last week
- Massively parallel rigidbody physics simulation on accelerator hardware.☆2,326Updated last week
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,673Updated last week
- An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly…☆7,302Updated this week
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.☆1,011Updated this week
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆614Updated this week
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,475Updated this week
- ☆2,689Updated 6 months ago
- Mastering Diverse Domains through World Models☆1,360Updated 3 months ago
- ♟️ Vectorized RL game environments in JAX☆405Updated this week
- JAX - A curated list of resources https://github.com/google/jax☆1,541Updated 3 months ago
- Flax is a neural network library for JAX that is designed for flexibility.☆6,107Updated this week
- High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, T…☆5,636Updated last month
- MuZero☆2,501Updated 2 months ago
- High throughput synchronous and asynchronous reinforcement learning☆823Updated 2 weeks ago
- The NetHack Learning Environment☆940Updated 6 months ago
- Implementation of Hinton's forward-forward (FF) algorithm - an alternative to back-propagation☆1,444Updated last year
- RL Environments in JAX 🌍☆641Updated 4 months ago
- Code for Cicero, an AI agent that plays the game of Diplomacy with open-domain natural language negotiation.☆1,302Updated last year
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.☆589Updated this week