ethanluoyc / corax
Corax: Core RL in JAX
☆36Updated 11 months ago
Alternatives and similar repositories for corax:
Users that are interested in corax are comparing it to the libraries listed below
- ☆47Updated 2 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆64Updated 8 months ago
- Skeleton for scalable and flexible Jax RL implementations☆71Updated last year
- Simple JAX Graphics Library.☆29Updated 3 months ago
- ☆35Updated 2 years ago
- ☆41Updated last year
- a modular reinforcement learning library with JAX agents☆22Updated last year
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆22Updated last year
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆55Updated 10 months ago
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆24Updated last year
- ☆70Updated 5 months ago
- EARL: Environment for Autonomous Reinforcement Learning☆36Updated 2 years ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆76Updated 2 months ago
- Highly scalable 2D JAX physics engine.☆44Updated 2 weeks ago
- Conservative Q learning in Jax☆52Updated 2 years ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Updated 9 months ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆109Updated 3 years ago
- ☆23Updated 2 years ago
- Modular Single-file Reinfocement Learning Algorithms Library☆37Updated last year
- ☆31Updated 10 months ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated last year
- Contains JAX implementation of algorithms for inverse reinforcement learning☆70Updated 5 months ago
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆60Updated last year
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆29Updated 6 months ago
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆12Updated 2 years ago
- Implementation of Tactical Optimistic and Pessimistic value estimation☆25Updated last year
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- ☆15Updated 9 months ago
- POPGym Library in JAX☆11Updated 10 months ago
- Foundation Policies with Hilbert Representations (ICML 2024)☆76Updated 10 months ago