enajx / NDP
☆52Updated 8 months ago
Alternatives and similar repositories for NDP:
Users that are interested in NDP are comparing it to the libraries listed below
- ☆37Updated 2 years ago
- ☆52Updated 4 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆92Updated 4 months ago
- Cellular Automata Accelerated in JAX (Oral at ICLR 2025).☆81Updated 2 months ago
- Efficient baselines for autocurricula in JAX.☆178Updated 5 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆70Updated 5 months ago
- ☆26Updated 8 months ago
- Official implementation of the DECKARD Agent from the paper "Do Embodied Agents Dream of Pixelated Sheep?"☆91Updated last year
- Code for the paper "Inference via Interpolation: Contrastive Representations Provably Enable Planning and Inference"☆40Updated 7 months ago
- ☆70Updated 5 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆48Updated 2 months ago
- ☆20Updated 7 months ago
- ☆32Updated 4 months ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆94Updated last year
- Generative cellular automaton-like learning environments for RL.☆19Updated 2 weeks ago
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).☆40Updated last month
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning☆63Updated this week
- An Open-Ended Agentic Simulator☆39Updated 6 months ago
- OMNI: Open-endedness via Models of human Notions of Interestingness☆42Updated 2 weeks ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated 4 months ago
- Implementation of Soft Actor Critic and some of its improvements in Pytorch☆52Updated this week
- ☆42Updated 7 months ago
- Solving the Abstraction & Reasoning Corpus with DreamCoder☆40Updated 4 months ago
- ☆73Updated 3 months ago
- ☆13Updated 6 months ago
- Intrinsic Motivation from Artificial Intelligence Feedback☆127Updated last year
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆27Updated last year
- The official repository for HyperZ⋅Z⋅W Operator Connects Slow-Fast Networks for Full Context Interaction.☆31Updated last week