subho406 / agalite
AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)
☆18Updated 6 months ago
Alternatives and similar repositories for agalite:
Users that are interested in agalite are comparing it to the libraries listed below
- ☆30Updated 5 months ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆15Updated 3 months ago
- Implementation of Proximal Policy Optimization in Jax+Flax☆19Updated last year
- ☆31Updated last year
- ☆27Updated last year
- Learning Robust Dynamics Through Variational Sparse Gating☆21Updated 2 years ago
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆28Updated 2 years ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆66Updated 11 months ago
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"☆22Updated 2 weeks ago
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆12Updated 4 years ago
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆44Updated this week
- General Modules for JAX☆64Updated last month
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 8 months ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- ☆27Updated 9 months ago
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆27Updated 9 months ago
- ☆18Updated 3 months ago
- Generalised UDRL☆37Updated 2 years ago
- ☆20Updated 2 years ago
- STABILIZING GRADIENTS FOR DEEP NEURAL NETWORKS VIA EFFICIENT SVD PARAMETERIZATION☆16Updated 6 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆21Updated this week
- GPT implementation in Flax☆18Updated 3 years ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆50Updated 5 months ago
- Implements the Messenger environment and EMMA model.☆23Updated last year
- flexible meta-learning in jax☆13Updated last year
- Expanding linear RNN state-transition matrix eigenvalues to include negatives improves state-tracking tasks and language modeling without…☆15Updated last month
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆56Updated 7 months ago
- POPGym Library in JAX☆11Updated last year
- An implementation of MuZero in JAX.☆56Updated 2 years ago