Popular GitHub Repositories for Brax

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

GPU-accelerated NeuroEvolution of Augmenting Topologies (NEAT)

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.

Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrenderer/wiki; PR: https://github.com/google/brax/pull/367

Walkthroughs for DSL, AirSim, the Vector Institute, and more

a modular reinforcement learning library with JAX agents

GPU-accelerated Evolutionary Multiobjective Optimization Using Tensorized RVEA.

Reinforcement learning in pure JAX.

An implementation of Short Horizon Actor Critic writen in Jax. Core algorithm written in the style of Brax, with several bits taken from Xu's original paper.

? Grasp - Pick-and-place with a robotic hand ??‍?

Simple implementations of Cartesian Genetic Programming (CGP) and Linear Genetic Programming (LGP) in JAX

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

Procedural Environment Generation for Accelerated Multi-Agent Reinforcement Learning