Egiob / cfrxLinks
cfrx is a collection of algorithms and tools for hardware-accelerated Counterfactual Regret Minimization (CFR) algorithms in Jax.
β37Updated last year
Alternatives and similar repositories for cfrx
Users that are interested in cfrx are comparing it to the libraries listed below
Sorting:
- β89Updated last year
- Baselines for gymnax π€β74Updated 2 years ago
- A collection of matrix games in JAXβ13Updated last year
- β91Updated 4 months ago
- πͺ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXβ61Updated 2 years ago
- A project that provides help for using DeepMind's mctx on gym-style environments.β63Updated last year
- Vectorization techniques for fast population-based training.β57Updated 3 years ago
- General Modules for JAXβ72Updated 3 months ago
- A collection of RL algorithms written in JAX.β104Updated 3 years ago
- Accelerated replay buffers in JAXβ46Updated 3 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weightsβ¦β59Updated 3 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!β254Updated 2 months ago
- Accelerated minigrid environments with JAXβ154Updated 2 months ago
- β‘ Flashbax: Accelerated Replay Buffers in JAXβ268Updated 3 months ago
- JAX implementations of core Deep RL algorithmsβ82Updated 3 years ago
- Simple single-file baselines for Q-Learning in pure-GPU settingβ231Updated last month
- Contains JAX implementation of algorithms for inverse reinforcement learningβ74Updated last year
- An implementation of MuZero in JAX.β58Updated 3 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancyβ22Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]β112Updated 2 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorchβ56Updated 2 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLβ120Updated last year
- IIG-RL-Benchmark is a library for training and evaluating game theoretical or deep RL algorithms on OpenSpiel games.β24Updated last month
- JAX implementation of RL algorithms and vectorized environmentsβ51Updated 2 years ago
- Classic MCTS example with mctxβ24Updated 2 years ago
- An Open-Ended Agentic Simulatorβ58Updated last year
- Code for Discovered Policy Optimisation (NeurIPS 2022)β12Updated 2 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselinβ¦β62Updated last week
- Scalable Opponent Shaping Experiments in JAXβ25Updated last year
- Clean single-file implementation of offline RL algorithms in JAXβ165Updated last month