enajx / NDP
☆53Updated 10 months ago
Alternatives and similar repositories for NDP:
Users that are interested in NDP are comparing it to the libraries listed below
- ☆38Updated 2 years ago
- ☆57Updated 6 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆97Updated 6 months ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆99Updated last year
- ☆29Updated last month
- MR.Q is a general-purpose model-free reinforcement learning algorithm.☆88Updated last week
- An Open-Ended Agentic Simulator☆46Updated 8 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆81Updated last month
- Cellular Automata Accelerated in JAX (Oral at ICLR 2025)☆86Updated this week
- ☆76Updated last month
- ☆20Updated 10 months ago
- Code for the paper "Harnessing Discrete Representations for Continual Reinforcement Learning"☆14Updated 10 months ago
- ☆19Updated 6 months ago
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).☆46Updated 3 months ago
- Efficient baselines for autocurricula in JAX.☆186Updated 7 months ago
- Highly scalable 2D JAX physics engine.☆54Updated last month
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆200Updated 2 weeks ago
- Simplest and Cleanest DreamerV3 implementation out there☆59Updated last month
- ☆74Updated 5 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆153Updated last month
- Reproduction of Dreamerv1 and v2 in pytorch for deepmind control suite☆38Updated 2 years ago
- Intrinsic Motivation from Artificial Intelligence Feedback☆128Updated last year
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆64Updated 10 months ago
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"☆22Updated 4 months ago
- Implementation of Soft Actor Critic and some of its improvements in Pytorch☆55Updated 2 months ago
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆174Updated 3 weeks ago
- Simple JAX Graphics Library.☆37Updated 5 months ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 3 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆72Updated 8 months ago
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆103Updated this week