ssokota / mmd
Code for magnetic mirror descent.
☆15Updated last year
Alternatives and similar repositories for mmd:
Users that are interested in mmd are comparing it to the libraries listed below
- Code for Model-Free Opponent Shaping (ICML 2022)☆16Updated 2 years ago
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"☆43Updated 7 months ago
- Scalable Opponent Shaping Experiments in JAX☆24Updated 10 months ago
- Results reproductions & comparisons between OpenSpiel implementations, associated paper & originating works☆17Updated 3 years ago
- Deep RL Code for XDO: A Double Oracle Algorithm for Extensive-Form Games☆37Updated 3 years ago
- ☆9Updated 3 years ago
- Modular Single-file Reinfocement Learning Algorithms Library☆37Updated last year
- A collection of matrix games in JAX☆9Updated 2 months ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆17Updated 3 months ago
- Novelty MiniGrid--NovGrid--is an extension of MiniGrid environment that allows for the world properties and dynamics to change according …☆35Updated 8 months ago
- Vectorization techniques for fast population-based training.☆55Updated 2 years ago
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆85Updated 3 years ago
- Conservative Q learning in Jax☆52Updated 2 years ago
- ☆35Updated 2 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆48Updated last year
- Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238☆44Updated 4 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆53Updated 2 years ago
- Synchronized Curriculum Learning for RL Agents☆33Updated this week
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆22Updated last year
- ☆41Updated last year
- Official Implementation for Quality-Similar Diversity via Population Based Reinforcement Learning☆17Updated last year
- Learning Task Embeddings for Teamwork Adaptation in Multi-Agent Reinforcement Learning☆12Updated 9 months ago
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- An Open-Ended Agentic Simulator☆39Updated 6 months ago
- Implementation of the Off Belief Learning algorithm.☆45Updated 2 years ago
- A collection of RL algorithms written in JAX.☆95Updated 2 years ago
- On the model-based stochastic value gradient for continuous reinforcement learning☆55Updated last year
- Mirror Descent Policy Optimization☆38Updated 4 years ago
- Corax: Core RL in JAX☆36Updated 11 months ago
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆12Updated 7 months ago