instadeepai / catx
đââŦ Contextual bandits library for continuous action trees with smoothing in JAX
â66Updated 2 years ago
Alternatives and similar repositories for catx:
Users that are interested in catx are comparing it to the libraries listed below
- Accelerated Quality-Diversityâ290Updated last week
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ234Updated last month
- đĒ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXâ57Updated last year
- A neural network library written in jaxâ11Updated 3 months ago
- A collection of matrix games in JAXâ11Updated 5 months ago
- A tool for aggregating and plotting MARL experiment data.â77Updated 3 months ago
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ322Updated this week
- JAX implementations of core Deep RL algorithmsâ79Updated 3 years ago
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papersâ51Updated 2 years ago
- â78Updated 6 months ago
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ724Updated this week
- Adaptable tools to make reinforcement learning and evolutionary computation algorithms.â56Updated 3 years ago
- Vectorization techniques for fast population-based training.â56Updated 2 years ago
- Recurrent and multi-process PyTorch implementation of deep reinforcement Actor-Critic algorithms A2C and PPOâ200Updated 2 years ago
- Baselines for gymnax đ¤â66Updated 2 years ago
- Benchmarking RL generalization in an interpretable way.â155Updated last month
- Partially Observable Process Gymâ188Updated 10 months ago
- Modular framework for Reinforcement Learning in pythonâ172Updated 2 years ago
- A collection of RL algorithms written in JAX.â97Updated 2 years ago
- Datasets with baselines for offline multi-agent reinforcement learning.â166Updated 2 weeks ago
- â101Updated last year
- â299Updated 4 months ago
- cfrx is a collection of algorithms and tools for hardware-accelerated Counterfactual Regret Minimization (CFR) algorithms in Jax.â31Updated 9 months ago
- JAX implementations of various deep reinforcement learning algorithms.â22Updated 3 months ago
- General Modules for JAXâ64Updated last month
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"â100Updated 2 years ago
- Accelerated minigrid environments with JAXâ135Updated 9 months ago
- Population-Based Reinforcement Learning for Combinatorial Optimizationâ73Updated last year
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the âĻâ85Updated 3 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!â205Updated last month