google / brax
Massively parallel rigidbody physics simulation on accelerator hardware.
☆2,498Updated this week
Alternatives and similar repositories for brax:
Users that are interested in brax are comparing it to the libraries listed below
- Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.☆3,902Updated this week
- A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.☆1,709Updated this week
- ☆1,280Updated this week
- Isaac Gym Reinforcement Learning Environments☆2,183Updated 3 months ago
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,118Updated 5 months ago
- Tiny Differentiable Simulator is a header-only C++ and CUDA physics library for reinforcement learning and robotics with zero dependencie…☆1,258Updated 3 months ago
- RL Environments in JAX 🌍☆682Updated 6 months ago
- ☆866Updated 7 months ago
- Library for Model Based RL☆979Updated 6 months ago
- Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Omniverse Isaac Gym and Isaac Lab☆630Updated this week
- RL implementations☆995Updated last month
- robosuite: A Modular Simulation Framework and Benchmark for Robot Learning☆1,445Updated this week
- Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research☆714Updated 2 weeks ago
- Open-source implementations of OpenAI Gym MuJoCo environments for use with the OpenAI Gym Reinforcement Learning Research Platform.☆838Updated 3 years ago
- A collection of robotics simulation environments for reinforcement learning☆617Updated last week
- A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents include…☆2,211Updated this week
- High throughput synchronous and asynchronous reinforcement learning☆860Updated 3 weeks ago
- Really Fast End-to-End Jax RL Implementations☆800Updated 4 months ago
- Code accompanying the paper "Learning Agile Robotic Locomotion Skills by Imitating Animals"☆1,213Updated last year
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆799Updated 5 months ago
- Reinforcement Learning Environments for Omniverse Isaac Gym☆916Updated 7 months ago
- Collections of robotics environments geared towards benchmarking multi-task and meta reinforcement learning☆1,314Updated 2 months ago
- A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.☆2,494Updated this week
- Mastering Diverse Domains through World Models☆1,464Updated 2 weeks ago
- JAX-based neural network library☆2,945Updated last month
- Multi-Joint dynamics with Contact. A general purpose physics simulator.☆8,626Updated this week
- A library of reinforcement learning components and agents☆3,570Updated 2 weeks ago
- Mastering Atari with Discrete World Models☆916Updated 2 years ago
- A collection of reference environments for offline reinforcement learning☆1,400Updated 2 months ago
- Robotics Gym Environments☆408Updated last year