#

brax

Here are 9 public repositories matching this topic...

EMI-Group / evox

Distributed GPU-Accelerated Framework for Evolutionary Computation. Comprehensive Library of Evolutionary Algorithms & Benchmark Problems.

neuroevolution pytorch gym gpu-acceleration evolutionary-algorithms multi-objective-optimization ray black-box-optimization evolutionary-computation evolutionary-strategies metaheuristics jax evolutionary-optimization derivative-free-optimization gradient-free-optimization population-based-optimization brax evolutionary-reinforcement-learinig

Updated Mar 5, 2025
Python

EMI-Group / tensorneat

GPU-accelerated NeuroEvolution of Augmenting Topologies (NEAT)

neat neuroevolution jax neuroevolutionofaugmentedtopologies brax neuroevolution-of-augmenting-topologies

Updated Feb 28, 2025
Python

perrin-isir / xpag

a modular reinforcement learning library with JAX agents

machine-learning reinforcement-learning deep-reinforcement-learning gym exploration flax sac jax td3 brax tqc sdqn

Updated Mar 3, 2025
Python

EMI-Group / metade

MetaDE is a GPU-accelerated evolutionary framework that optimizes Differential Evolution (DE) strategies via meta-level evolution. Supporting both JAX and PyTorch, it dynamically adapts mutation and crossover strategies for efficient large-scale black-box optimization.

neuroevolution pytorch evolutionary-algorithms gpu-computing differential-evolution jax brax

Updated Mar 4, 2025
Python

EMI-Group / tensorrvea

GPU-accelerated Evolutionary Multiobjective Optimization Using Tensorized RVEA.

neuroevolution evolutionary-algorithms gpu-computing multi-objective-optimization jax brax

Updated Feb 25, 2025
Python

rystrauss / dopamax

Reinforcement learning in pure JAX.

reinforcement-learning dqn mcts ddpg sac ppo podracer alphazero jax td3 muzero brax anakin dopamax

Updated Feb 23, 2025
Python

tartavull / alfredo

Relentlessly learning, persistently failing, but never surrendering.

reinforcement-learning artificial-intelligence evolutionary-algorithms genetic-algorithms gym-environments brax

Updated Dec 1, 2024
Python

alpc91 / SMERL

[ICML 2024] Official environments and JAX-implementations for "Subequivariant Reinforcement Learning in 3D Multi-Entity Physical Environments"

reinforcement-learning deep-learning graph multi-agent symmetry geometric-deep-learning jax brax

Updated Jul 18, 2024
Python

tomalner18 / MAax

Procedural Environment Generation for Accelerated Multi-Agent Reinforcement Learning

python reinforcement-learning procedural-generation physics-simulation massively-parallel multiagent-reinforcement-learning jax reinforcement-learning-environments brax

Updated May 29, 2024
Python

Improve this page

Add a description, image, and links to the brax topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the brax topic, visit your repo's landing page and select "manage topics."