alexlioralexli / learned-fourier-features
Code for the paper "Functional Regularization for Reinforcement Learning via Learned Fourier Features"
☆16Updated last year
Related projects: ⓘ
- Vectorization techniques for fast population-based training.☆52Updated 2 years ago
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆83Updated 3 years ago
- Supplementary Data for Evolving Reinforcement Learning Algorithms☆46Updated 3 years ago
- Reinforcement Learning with Latent Flow☆42Updated 3 years ago
- Baselines for gymnax 🤖☆57Updated last year
- Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"☆39Updated last year
- Code repo for Gradient Temporal-Difference Learning with Regularized Corrections paper.☆35Updated 3 years ago
- Deep Reinforcement Learning Framework done with PyTorch☆27Updated 3 weeks ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆49Updated 2 years ago
- ☆11Updated this week
- My Body Is A Cage☆37Updated 3 years ago
- Foundation Policies with Hilbert Representations (ICML 2024)☆65Updated 5 months ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆31Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆76Updated last year
- Code release for Efficient Planning in a Compact Latent Action Space (ICLR2023) https://arxiv.org/abs/2208.10291.☆90Updated last year
- Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observ…☆25Updated last year
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆49Updated 11 months ago
- On the model-based stochastic value gradient for continuous reinforcement learning☆54Updated last year
- A JAX Implementation of the Twin Delayed DDPG Algorithm☆31Updated 4 years ago
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆25Updated 10 months ago
- Proto-RL: Reinforcement Learning with Prototypical Representations☆81Updated 2 years ago
- General Modules for JAX☆57Updated last month
- Pytorch implementation of DreamerV2: MASTERING ATARI WITH DISCRETE WORLD MODELS☆49Updated 2 years ago
- Generalised UDRL☆37Updated 2 years ago
- Intrinsic Reward Matching (IRM) implementation (from Adeniji and Xie et al 2022)☆43Updated 8 months ago
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆49Updated 8 months ago
- ☆34Updated last year
- MELD: Meta-Reinforcement Learning from Images via Latent State Models https://arxiv.org/abs/2010.13957☆62Updated 3 years ago
- AGAC: Adversarially Guided Actor-Critic☆47Updated 3 years ago
- Official pytorch implementation for our ICLR 2023 paper "Latent State Marginalization as a Low-cost Approach for Improving Exploration".☆24Updated last year