RL Environments in JAX 🌍

Low cost motion capture system for room scale tracking

A bare-bones Python library for quality diversity optimization.

A simple Nash Equilibrium solver for two-player zero-sum games

The autonomous racing stack for the ForzaETH team at PBL

OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.

Multi-Agent Reinforcement Learning with JAX

A JAX-based simulator for autonomous driving research.

Accelerated Quality-Diversity

[ICCV & CVPR Workshop] Learning-enabled Interactive Prediction and Planning Framework for Autonomous Vehicles

[CoRL'23] Parting with Misconceptions about Learning-based Vehicle Motion Planning

Hardware accelerated, batchable and differentiable optimizers in JAX.

The entrance repository of Markdown presentation ecosystem

JAX - A curated list of resources https://github.com/google/jax

Differentiable convex optimization layers

Massively parallel rigidbody physics simulation on accelerator hardware.

Implementation of the Paper "Permutation-Invariant Variational Autoencoder for Graph-Level Representation Learning" by Robin Winter, Frank Noe and Djork-Arne Clevert.

A versatile driver for a wide range of SICK LiDAR and RADAR devices, providing support for both Linux (native, ROS 1, ROS 2) and Windows (native, ROS 2) platforms.

A LiDAR odometry pipeline that just works

ROS visualisation on the terminal

VESC EDU mounting plate for Traxxas Slash LCG chassis

Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.

Chemical-Reaction-Aware Molecule Representation Learning

Code snippets for ROS2

Optax is a gradient processing and optimization library for JAX.

Evolution Strategies in JAX 🦎

Multi-Joint dynamics with Contact. A general purpose physics simulator.