tristandeleu / jax-meta-learningLinks
A collection of meta-learning algorithms in Jax
☆23Updated 2 years ago
Alternatives and similar repositories for jax-meta-learning
Users that are interested in jax-meta-learning are comparing it to the libraries listed below
Sorting:
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Updated 4 years ago
- flexible meta-learning in jax☆14Updated last year
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- PyTorch Package For Quasimetric Learning☆42Updated 7 months ago
- ☆22Updated 2 months ago
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- ☆56Updated 2 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆19Updated 4 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆56Updated 2 years ago
- An implementation of MuZero in JAX.☆56Updated 2 years ago
- ☆28Updated 2 years ago
- ☆44Updated 9 months ago
- General Modules for JAX☆65Updated 2 months ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆18Updated 7 months ago
- Sandbox environment for generalizable agent research☆25Updated 2 years ago
- Variational Reinforcement Learning☆16Updated 11 months ago
- Dreamer on JAX☆16Updated 3 years ago
- JAX implementations of core Deep RL algorithms☆79Updated 3 years ago
- Vectorization techniques for fast population-based training.☆56Updated 2 years ago
- ☆31Updated 4 years ago
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆28Updated 2 years ago
- Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"☆44Updated last year
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"☆48Updated 11 months ago
- Generalised UDRL☆37Updated 3 years ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆15Updated 2 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- Continual Reinforcement Learning in 3D Non-stationary Environments☆38Updated 6 years ago
- POPGym Library in JAX☆11Updated last year
- Flax Implementation of DreamerV3 on Crafter☆16Updated 3 months ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆17Updated 2 years ago