QDax: A Library for Quality-Diversity and Population-based Algorithms with Hardware Acceleration
Not on the map. Why?
Abstract (arXiv)
QDax is an open-source library with a streamlined and modular API for Quality-Diversity (QD) optimization algorithms in Jax. The library serves as a versatile tool for optimization purposes, ranging from black-box optimization to continuous control. QDax offers implementations of popular QD, Neuroevolution, and Reinforcement Learning (RL) algorithms, supported by various examples. All the implementations can be just-in-time compiled with Jax, facilitating efficient execution across multiple accelerators, including GPUs and TPUs. These implementations effectively demonstrate the framework's flexibility and user-friendliness, easing experimentation for research purposes. Furthermore, the library is thoroughly documented and tested with 95% coverage.
Key facts. Cited by 16 papers in the map (data through 7 Oct 2026). Builds on 26 of its 35 references that are in the map. First posted on arXiv 7 Aug 2023.
Builds on
26 of 35 references.
References · 26
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2021Isaac Gym: High Performance GPU-Based Physics Simulation For Robot Learningcited by 882
- 2018Diversity is All You Need: Learning Skills without a Reward Functioncited by 734
- 2015Illuminating search spaces by mapping elitescited by 406
- 2021Brax -- A Differentiable Physics Engine for Large Scale Rigid Body Simulationcited by 375
- 2017Population Based Training of Neural Networkscited by 351
- 2019Dynamics-Aware Unsupervised Discovery of Skillscited by 308
- 2014Robots that can adapt like animalscited by 203
- 2020One Solution is Not All You Need: Few-Shot Extrapolation via Structured MaxEnt RLcited by 79
- 2019Covariance Matrix Adaptation for the Rapid Illumination of Behavior Spacecited by 78
- 2021Differentiable Quality Diversitycited by 71
- 2016Using Centroidal Voronoi Tessellations to Scale Up the Multi-dimensional Archive of Phenotypic Elites Algorithmcited by 65
14 more
- 2020Diversity Policy Gradient for Sample Efficient Quality-Diversity Optimizationcited by 53
- 2023Jumanji: a Diverse Suite of Scalable Reinforcement Learning Environments in JAXcited by 52
- 2020Scaling MAP-Elites to Deep Neuroevolutioncited by 38
- 2022EvoJAX: Hardware-Accelerated Neuroevolutioncited by 29
- 2023pyribs: A Bare-Bones Python Library for Quality Diversity Optimizationcited by 29
- 2022Neuroevolution is a Competitive Alternative to Reinforcement Learning for Skill Discoverycited by 23
- 2019Empowering Quality Diversity in Dungeon Design with Interactive Constrained MAP-Elitescited by 18
- 2022Multi-Objective Quality Diversity Optimizationcited by 18
- 2022Empirical analysis of PGA-MAP-Elites for Neuroevolution in Uncertain Domainscited by 15
- 2022Accelerated Quality-Diversity through Massive Parallelismcited by 14
- 2022Benchmarking Quality-Diversity Algorithms on Neuroevolution for Reinforcement Learningcited by 10
- 2017Data-Efficient Exploration, Optimization, and Modeling of Diverse Designs through Surrogate-Assisted Illuminationcited by 6
- 2023Evolving Populations of Diverse RL Agents with MAP-Elitescited by 5
- 2022Assessing Quality-Diversity Neuro-Evolution Algorithms Performance in Hard Exploration Problemscited by 4
Built on by
Citing papers per year: 2023: 3; 2024: 3; 2025: 6; 2026: 4.
16 papers.
Sort:
By area:
- 2024CAX: Cellular Automata Accelerated in JAXcited by 8
- 2025Dominated Novelty Search: Rethinking Local Competition in Quality-Diversitycited by 7
- 2023Synergizing Quality-Diversity with Descriptor-Conditioned Reinforcement Learningcited by 6
- 2023Diversity from Human Feedbackcited by 5
- 2023Mix-ME: Quality-Diversity for Multi-Agent Learningcited by 4
- 2025Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimizationcited by 2
- 2025Multi-Objective Quality-Diversity in Unstructured and Unbounded Spacescited by 2
- 2025Breaking the Performance Ceiling in Reinforcement Learning requires Inference Strategiescited by 2
- 2025When Does Neuroevolution Outcompete Reinforcement Learning in Transfer Learning Tasks?cited by 2
- 2024Quality-Diversity with Limited Resourcescited by 1
- 2024Preference-Conditioned Gradient Variations for Multi-Objective Quality-Diversitycited by 1
- 2026Crazyflow: An Accurate, GPU-Accelerated, Differentiable Drone Simulator in JAXcited by 1
4 more
- 2025Scaling Policy Gradient Quality-Diversity with Massive Parallelization via Behavioral Variationscited by 0
- 2026Distributional Value Estimation Without Target Networks for Robust Quality-Diversitycited by 0
- 2026Contextual Quality-Diversity Evolutionary Reinforcement Learning for HVAC Control in Tropical Commercial Buildingscited by 0
- 2026Behavioral Foundation Models for Quality Diversitycited by 0
- 2026Behavioral Foundation Models for Quality Diversity28 Sep 2026
- 2026Contextual Quality-Diversity Evolutionary Reinforcement Learning for HVAC Control in Tropical Commercial Buildings11 Aug 2026
- 2026Crazyflow: An Accurate, GPU-Accelerated, Differentiable Drone Simulator in JAX31 May 2026
- 2026Distributional Value Estimation Without Target Networks for Robust Quality-Diversity22 Apr 2026
- 2025When Does Neuroevolution Outcompete Reinforcement Learning in Transfer Learning Tasks?28 May 2025
- 2025Breaking the Performance Ceiling in Reinforcement Learning requires Inference Strategies27 May 2025
- 2025Multi-Objective Quality-Diversity in Unstructured and Unbounded Spaces28 Mar 2025
- 2025Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimization4 Feb 2025
- 2025Dominated Novelty Search: Rethinking Local Competition in Quality-Diversity1 Feb 2025
- 2025Scaling Policy Gradient Quality-Diversity with Massive Parallelization via Behavioral Variations30 Jan 2025
- 2024Preference-Conditioned Gradient Variations for Multi-Objective Quality-Diversity19 Nov 2024
- 2024CAX: Cellular Automata Accelerated in JAX3 Oct 2024
4 more
- 2024Quality-Diversity with Limited Resources6 Jun 2024
- 2023Synergizing Quality-Diversity with Descriptor-Conditioned Reinforcement Learning10 Dec 2023
- 2023Mix-ME: Quality-Diversity for Multi-Agent Learning3 Nov 2023
- 2023Diversity from Human Feedback10 Oct 2023
- 2023Diversity from Human Feedbackcited by 5
- 2023Diversity from Human Feedback10 Oct 2023
- 2024CAX: Cellular Automata Accelerated in JAXcited by 8
- 2025Dominated Novelty Search: Rethinking Local Competition in Quality-Diversitycited by 7
- 2023Synergizing Quality-Diversity with Descriptor-Conditioned Reinforcement Learningcited by 6
- 2023Mix-ME: Quality-Diversity for Multi-Agent Learningcited by 4
- 2025Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimizationcited by 2
- 2025Multi-Objective Quality-Diversity in Unstructured and Unbounded Spacescited by 2
- 2025Breaking the Performance Ceiling in Reinforcement Learning requires Inference Strategiescited by 2
- 2025When Does Neuroevolution Outcompete Reinforcement Learning in Transfer Learning Tasks?cited by 2
- 2024Quality-Diversity with Limited Resourcescited by 1
- 2024Preference-Conditioned Gradient Variations for Multi-Objective Quality-Diversitycited by 1
- 2026Crazyflow: An Accurate, GPU-Accelerated, Differentiable Drone Simulator in JAXcited by 1
- 2025Scaling Policy Gradient Quality-Diversity with Massive Parallelization via Behavioral Variationscited by 0
12 most cited of 15 All 15 (JSON)
- 2026Behavioral Foundation Models for Quality Diversity28 Sep 2026
- 2026Contextual Quality-Diversity Evolutionary Reinforcement Learning for HVAC Control in Tropical Commercial Buildings11 Aug 2026
- 2026Crazyflow: An Accurate, GPU-Accelerated, Differentiable Drone Simulator in JAX31 May 2026
- 2026Distributional Value Estimation Without Target Networks for Robust Quality-Diversity22 Apr 2026
- 2025When Does Neuroevolution Outcompete Reinforcement Learning in Transfer Learning Tasks?28 May 2025
- 2025Breaking the Performance Ceiling in Reinforcement Learning requires Inference Strategies27 May 2025
- 2025Multi-Objective Quality-Diversity in Unstructured and Unbounded Spaces28 Mar 2025
- 2025Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimization4 Feb 2025
- 2025Dominated Novelty Search: Rethinking Local Competition in Quality-Diversity1 Feb 2025
- 2025Scaling Policy Gradient Quality-Diversity with Massive Parallelization via Behavioral Variations30 Jan 2025
- 2024Preference-Conditioned Gradient Variations for Multi-Objective Quality-Diversity19 Nov 2024
- 2024CAX: Cellular Automata Accelerated in JAX3 Oct 2024
12 newest of 15 All 15 (JSON)