Code for "Baba Is AI: Break the Rules to Beat the Benchmark"
☆42Sep 3, 2025Updated 6 months ago
Alternatives and similar repositories for baba-is-ai
Users that are interested in baba-is-ai are comparing it to the libraries listed below
Sorting:
- Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to R…☆12Feb 23, 2024Updated 2 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆23Oct 28, 2024Updated last year
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25May 5, 2024Updated last year
- Learning diverse options through the Laplacian representation.☆23Jan 5, 2024Updated 2 years ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Oct 12, 2023Updated 2 years ago
- ☆10Jun 27, 2024Updated last year
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Apr 19, 2024Updated last year
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆17Nov 24, 2025Updated 3 months ago
- Pointax: PointMaze Environment for JAX☆26Oct 22, 2025Updated 4 months ago
- ☆16Jul 16, 2024Updated last year
- [ICML 2025 GenBio Workshop] Official Implementation for "Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentia…☆17Jun 12, 2025Updated 8 months ago
- This is the official code implementation of Bongard-OpenWorld (ICLR 2024).☆14Jan 6, 2025Updated last year
- Dual optimization to learn laplacian eigenpairs in arbitrary spaces☆16Dec 18, 2024Updated last year
- A toolkit for practical Human-AI cooperation research☆14Apr 19, 2024Updated last year
- Measuring the Signal to Noise Ratio in Language Model Evaluation☆28Aug 19, 2025Updated 6 months ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆26Jan 14, 2025Updated last year
- Official implementation of "Latent Action Learning Requires Supervision in the Presence of Distractors", ICML 2025☆33Jul 8, 2025Updated 8 months ago
- Official codebase for "Sampling For Learnability", published at NeurIPS 2024☆20Oct 21, 2025Updated 4 months ago
- ☆19May 20, 2025Updated 9 months ago
- ☆18Jul 4, 2025Updated 8 months ago
- ☆18Aug 20, 2025Updated 6 months ago
- Parallel Associative Scan for Language Models☆18Jan 8, 2024Updated 2 years ago
- ☆19Apr 22, 2024Updated last year
- Gym environment pybullet based for simulating a quadruped (optionally) PEA equipped.☆22Mar 3, 2025Updated last year
- High-performance JAX-powered simulator for robotic navigation in 2D mazes, optimized for Quality-Diversity algorithm research and benchma…☆20Jun 19, 2025Updated 8 months ago
- Official pytorch implementation for our ICLR 2023 paper "Latent State Marginalization as a Low-cost Approach for Improving Exploration".☆24Feb 9, 2023Updated 3 years ago
- ☆44Jul 12, 2024Updated last year
- A dataloader, but for JAX☆20May 17, 2024Updated last year
- Scalable Opponent Shaping Experiments in JAX☆25Apr 13, 2024Updated last year
- Aria Training and Evaluation Kit☆32Jan 23, 2026Updated last month
- Official code for Slot-Transformer for Videos (STEVE)☆50Jan 9, 2023Updated 3 years ago
- ☆24Jul 8, 2023Updated 2 years ago
- Multi-level Online Sequential Experts (MOSE) for online continual learning problem. (CVPR2024)☆24Nov 21, 2025Updated 3 months ago
- The code corresponding to the paper "Improving Sample Efficiency of Deep Reinforcement Learning for Bipedal Walking".☆24Aug 8, 2022Updated 3 years ago
- Scripts to recreate the D4RL datasets with Minari☆26Jul 21, 2025Updated 7 months ago
- [ICLR 2024 oral] Pre-Training Goal-based Models for Sample-Efficient Reinforcement Learning☆29Mar 1, 2024Updated 2 years ago
- ☆19Nov 25, 2022Updated 3 years ago
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆25Oct 14, 2024Updated last year
- minimal Energy-based transformer☆43Dec 11, 2025Updated 2 months ago