πββ¬ 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