This project was moved to: https://github.com/coax-dev/coax
☆161Nov 28, 2022Updated 3 years ago
Alternatives and similar repositories for coax
Users that are interested in coax are comparing it to the libraries listed below
Sorting:
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆493Apr 6, 2024Updated last year
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 4 years ago
- Tonic RL library☆419Jul 24, 2024Updated last year
- ☆1,407Dec 9, 2025Updated 2 months ago
- Ranking Policy Gradient☆23Nov 27, 2019Updated 6 years ago
- Modular framework for Reinforcement Learning in python☆183Feb 1, 2023Updated 3 years ago
- Solving reinforcement learning tasks which require language and vision☆33Apr 4, 2023Updated 2 years ago
- ☆774Jan 27, 2024Updated 2 years ago
- A collection of RL algorithms written in JAX.☆105Jul 5, 2022Updated 3 years ago
- ☆126Aug 9, 2023Updated 2 years ago
- ☆620Jan 23, 2026Updated last month
- ☆84Nov 19, 2020Updated 5 years ago
- A JAX implementation of stochastic addition.☆14Aug 15, 2022Updated 3 years ago
- 📴 OffCon^3: SOTA PyTorch SAC and TD3 Implementations (arxiv: 2101.11331)☆25Jun 20, 2021Updated 4 years ago
- A High Level API for Deep Learning in JAX☆476Dec 15, 2022Updated 3 years ago
- PFRL: a PyTorch-based deep reinforcement learning library☆1,261Dec 15, 2025Updated 2 months ago
- ☆54Jan 13, 2023Updated 3 years ago
- ROS package for robot learning☆17Oct 16, 2019Updated 6 years ago
- DeepMind Alchemy task environment: a meta-reinforcement learning benchmark☆205Apr 5, 2023Updated 2 years ago
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent☆1,529Apr 13, 2024Updated last year
- A PyTorch reinforcement learning library for generalizable and reproducible algorithm implementations with an aim to improve accessibilit…☆413Dec 27, 2022Updated 3 years ago
- A library of reinforcement learning components and agents☆3,927Feb 16, 2026Updated last week
- Optax is a gradient processing and optimization library for JAX.☆2,193Updated this week
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46May 4, 2021Updated 4 years ago
- ☆925Jan 29, 2026Updated last month
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Jul 19, 2025Updated 7 months ago
- Normalizing Flows using JAX☆86Nov 30, 2023Updated 2 years ago
- Exponential families for JAX☆76Feb 20, 2026Updated last week
- Convert DeepMind Control Suite to OpenAI gym environments.☆87Jan 31, 2020Updated 6 years ago
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated 11 months ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆516Updated this week
- ☆164Dec 13, 2023Updated 2 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆191Sep 28, 2022Updated 3 years ago
- lagom: A PyTorch infrastructure for rapid prototyping of reinforcement learning algorithms.☆378Nov 19, 2022Updated 3 years ago
- OpenAI gym environment for evolving morphologies of 2D virtual creatures.☆34Jul 26, 2023Updated 2 years ago
- A library that makes Evolutionary Strategies (ES) simple to use.☆179Apr 14, 2021Updated 4 years ago
- A simple OpenAI Gym environment for single and multi-agent reinforcement learning☆776Dec 14, 2023Updated 2 years ago
- Gym implementation of connector to Deepmind lab☆12Mar 26, 2019Updated 6 years ago
- JAX implementations of core Deep RL algorithms☆83May 2, 2022Updated 3 years ago