🐈⬛ Contextual bandits library for continuous action trees with smoothing in JAX
☆71Oct 7, 2022Updated 3 years ago
Alternatives and similar repositories for catx
Users that are interested in catx are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Accelerated Quality-Diversity☆345Oct 30, 2025Updated 4 months ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆817Mar 9, 2026Updated 2 weeks ago
- 🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX☆887Feb 26, 2026Updated 3 weeks ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆61Oct 23, 2023Updated 2 years ago
- Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Function☆13Nov 22, 2022Updated 3 years ago
- LightMHC: A Light Model for pMHC Structure Prediction with Graph Neural Networks☆12Jan 9, 2024Updated 2 years ago
- Population-Based Reinforcement Learning for Combinatorial Optimization☆87Feb 12, 2024Updated 2 years ago
- Adapting the AlphaZero algorithm to remove the need of execution traces to train NPI.☆80Oct 3, 2023Updated 2 years ago
- Vectorization techniques for fast population-based training.☆57Aug 12, 2022Updated 3 years ago
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papers☆56Jan 20, 2023Updated 3 years ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- A collection of matrix games in JAX☆13Nov 28, 2024Updated last year
- Estimators to perform off-policy evaluation☆13Sep 3, 2024Updated last year
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆397Mar 1, 2026Updated 3 weeks ago
- Black-box Bayesian inference for agent-based models☆32Aug 25, 2024Updated last year
- A first bare bones paralleled implementation of Go Explore as described by the Uber Engineering blog post☆46Jan 25, 2019Updated 7 years ago
- Library for efficient training and application of Machine Learning Interatomic Potentials (MLIP)☆84Jan 7, 2026Updated 2 months ago
- Revisiting Discrete Gradient Estimation in MADDPG☆27Feb 24, 2023Updated 3 years ago
- JAX/Flax implementation of the Hyena Hierarchy☆34Apr 27, 2023Updated 2 years ago
- A LLM-friendly framework for translating dynamical equations to gymnasium-compatible RL environments.☆33Updated this week
- ☆24Jul 31, 2023Updated 2 years ago
- Continual learning strategies(EWC, GEM) for rotated MNIST dataset☆12Apr 6, 2020Updated 5 years ago
- ☆11Dec 23, 2025Updated 3 months ago
- JuMP extensions for constraint programming.☆15Jun 5, 2020Updated 5 years ago
- Code for "Task-Agnostic Continual RL: In Praise of a Simple Baseline"☆35Jun 8, 2023Updated 2 years ago
- This repository demonstrates the application of our proposed task-free continual learning method on a synthetic experiment.☆13Jun 24, 2019Updated 6 years ago
- jax-triton contains integrations between JAX and OpenAI Triton☆442Mar 13, 2026Updated last week
- Orbax provides common checkpointing and persistence utilities for JAX users☆490Updated this week
- COMPASS: Combinatorial Optimization with Policy Adaptation using Latent Space Search☆42Jun 21, 2024Updated last year
- Qt-like event loops, signals and slots for communication across threads and processes in Python☆14Mar 26, 2024Updated last year
- OpenAI Gym environment for graph search problems such as shortest path.☆11Dec 24, 2019Updated 6 years ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆108Nov 25, 2025Updated 3 months ago
- ☆14Feb 24, 2025Updated last year
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆122Aug 22, 2024Updated last year
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- OpenRAVE Controller Plugin for UR5 (Universal Robots UR5) Robot☆33Jun 22, 2020Updated 5 years ago
- Reinforcement learning in pure JAX.☆13Dec 24, 2025Updated 2 months ago
- Simulation Streams is a programming paradigm designed to efficiently control and leverage Large Language Models (LLMs) for complex, dynam…☆24Mar 13, 2025Updated last year
- (ICML2022) Off-Policy Evaluation for Large Action Spaces via Embeddings☆21Jul 27, 2022Updated 3 years ago