quasimetric-learning / torch-quasimetric
PyTorch Package For Quasimetric Learning
☆42Updated last week
Related projects ⓘ
Alternatives and complementary repositories for torch-quasimetric
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆19Updated 5 months ago
- Building blocks for productive research☆44Updated 2 weeks ago
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆26Updated 2 years ago
- Generalised UDRL☆37Updated 2 years ago
- ☆36Updated last year
- ☆56Updated 2 years ago
- ☆25Updated 11 months ago
- Accelerated replay buffers in JAX☆39Updated 2 years ago
- Model-Based Visual Planning with Self-Supervised Functional Distances (ICLR 2021)☆20Updated 3 years ago
- General Modules for JAX☆58Updated 3 months ago
- ☆28Updated 3 years ago
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆25Updated last year
- Proto-RL: Reinforcement Learning with Prototypical Representations☆82Updated 2 years ago
- GPT implementation in Flax☆18Updated 2 years ago
- Variational Reinforcement Learning☆16Updated 3 months ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆52Updated 7 months ago
- ☆15Updated 2 years ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆83Updated 9 months ago
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆79Updated last year
- ☆53Updated this week
- Code repository complementing the ICLR 2021 paper "Unsupervised Object Keypoint Learning using Local Spatial Predictability" (https://arx…☆9Updated 3 years ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 3 months ago
- My Body Is A Cage☆38Updated 3 years ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆32Updated last week
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆30Updated last year
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Updated 3 years ago
- ☆22Updated 4 months ago
- ☆41Updated last month