Howuhh / sac-n-jax
Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch
☆46Updated last year
Related projects ⓘ
Alternatives and complementary repositories for sac-n-jax
- Accelerated replay buffers in JAX☆39Updated 2 years ago
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆66Updated last year
- ☆16Updated 3 months ago
- An Open-Ended Agentic Simulator☆22Updated 2 months ago
- Code for the paper "Showing Your Offline Reinforcement Learning Work: Online Evaluation Budget Matters", ICML 2022☆28Updated 2 years ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆79Updated last year
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆37Updated last year
- Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC☆52Updated last year
- ☆61Updated 2 months ago
- Simple JAX Graphics Library.☆13Updated this week
- Challenging Memory-based Deep Reinforcement Learning Agents☆84Updated last week
- Vectorization techniques for fast population-based training.☆54Updated 2 years ago
- Conservative Q learning in Jax☆50Updated last year
- Reinforcement learning on general 2D physics environments in JAX☆10Updated this week
- Official implementation for "Q-Ensemble for Offline RL: Don't Scale the Ensemble, Scale the Batch Size", NeurIPS 2022, Offline RL Worksho…☆21Updated last year
- ☆37Updated last year
- Baselines for gymnax 🤖☆58Updated last year
- ☆42Updated last year
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- General Modules for JAX☆58Updated 3 months ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆57Updated 5 months ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆98Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆49Updated 2 years ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 2 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆104Updated 2 years ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆47Updated last year
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆21Updated 6 months ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆40Updated last week
- ☆36Updated last year
- On the model-based stochastic value gradient for continuous reinforcement learning☆55Updated last year