Egiob / cfrxLinks
cfrx is a collection of algorithms and tools for hardware-accelerated Counterfactual Regret Minimization (CFR) algorithms in Jax.
β36Updated last year
Alternatives and similar repositories for cfrx
Users that are interested in cfrx are comparing it to the libraries listed below
Sorting:
- Vectorization techniques for fast population-based training.β56Updated 3 years ago
- Baselines for gymnax π€β73Updated 2 years ago
- A project that provides help for using DeepMind's mctx on gym-style environments.β63Updated last year
- β88Updated last year
- Accelerated replay buffers in JAXβ45Updated 3 years ago
- πͺ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXβ60Updated 2 years ago
- A collection of matrix games in JAXβ12Updated last year
- β89Updated 3 months ago
- Scalable Opponent Shaping Experiments in JAXβ24Updated last year
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorchβ56Updated 2 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!β251Updated last month
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weightsβ¦β59Updated 3 years ago
- A collection of RL algorithms written in JAX.β104Updated 3 years ago
- General Modules for JAXβ71Updated 3 months ago
- Code for magnetic mirror descent.β16Updated 2 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancyβ22Updated last year
- JAX implementations of core Deep RL algorithmsβ82Updated 3 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLβ118Updated last year
- β‘ Flashbax: Accelerated Replay Buffers in JAXβ264Updated 2 months ago
- fast + parallel AlphaZero in JAXβ107Updated 11 months ago
- Accelerated minigrid environments with JAXβ153Updated 2 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)β12Updated 2 years ago
- An implementation of MuZero in JAX.β57Updated 3 years ago
- IIG-RL-Benchmark is a library for training and evaluating game theoretical or deep RL algorithms on OpenSpiel games.β23Updated last month
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the β¦β92Updated 4 years ago
- Simple single-file baselines for Q-Learning in pure-GPU settingβ229Updated 3 weeks ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]β111Updated 2 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselinβ¦β62Updated 3 weeks ago
- Reinforcement learning in pure JAX.β13Updated 9 months ago
- Challenging Memory-based Deep Reinforcement Learning Agentsβ105Updated last year