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:
- A pathway and collection of resources to learning Jax from beginning to advance.☆11Jan 2, 2021Updated 5 years ago
- Modular framework for Reinforcement Learning in python☆184Feb 1, 2023Updated 3 years ago
- ☆1,411Mar 2, 2026Updated 2 weeks ago
- ☆774Jan 27, 2024Updated 2 years ago
- ☆627Jan 23, 2026Updated last month
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆495Apr 6, 2024Updated last year
- Tonic RL library☆419Jul 24, 2024Updated last year
- A JAX implementation of stochastic addition.☆14Aug 15, 2022Updated 3 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 4 years ago
- Ranking Policy Gradient☆23Nov 27, 2019Updated 6 years ago
- Solving reinforcement learning tasks which require language and vision☆33Apr 4, 2023Updated 2 years ago
- Revisiting Rainbow☆76Jun 9, 2021Updated 4 years ago
- A High Level API for Deep Learning in JAX☆476Dec 15, 2022Updated 3 years ago
- ☆929Mar 12, 2026Updated last week
- ☆84Nov 19, 2020Updated 5 years ago
- Optax is a gradient processing and optimization library for JAX.☆2,212Updated this week
- A collection of RL algorithms written in JAX.☆105Jul 5, 2022Updated 3 years ago
- A library of reinforcement learning components and agents☆3,946Mar 12, 2026Updated last week
- 📴 OffCon^3: SOTA PyTorch SAC and TD3 Implementations (arxiv: 2101.11331)☆25Jun 20, 2021Updated 4 years ago
- DeepMind Alchemy task environment: a meta-reinforcement learning benchmark☆205Apr 5, 2023Updated 2 years ago
- PFRL: a PyTorch-based deep reinforcement learning library☆1,266Mar 2, 2026Updated 2 weeks ago
- ☆54Jan 13, 2023Updated 3 years ago
- ROS package for robot learning☆17Oct 16, 2019Updated 6 years ago
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent☆1,531Apr 13, 2024Updated last year
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46May 4, 2021Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Jul 19, 2025Updated 8 months ago
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated last year
- Exponential families for JAX☆76Mar 13, 2026Updated last week
- Gym environments for Robots that learn to interact with the environment autonomously☆34Dec 26, 2022Updated 3 years ago
- A PyTorch reinforcement learning library for generalizable and reproducible algorithm implementations with an aim to improve accessibilit…☆412Dec 27, 2022Updated 3 years ago
- JAX-based neural network library☆3,197Mar 12, 2026Updated last week
- lagom: A PyTorch infrastructure for rapid prototyping of reinforcement learning algorithms.☆378Nov 19, 2022Updated 3 years ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆519Mar 2, 2026Updated 2 weeks ago
- Normalizing Flows using JAX☆87Nov 30, 2023Updated 2 years ago
- A PyTorch Platform for Distributed RL☆751Sep 15, 2021Updated 4 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆191Sep 28, 2022Updated 3 years ago
- Concise deep learning for JAX☆183Oct 21, 2020Updated 5 years ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆330Mar 20, 2024Updated 2 years ago
- ☆164Dec 13, 2023Updated 2 years ago