brax

Physics simulator

A physics simulation framework designed for research and development in robotics, reinforcement learning, and other fields.

Massively parallel rigidbody physics simulation on accelerator hardware.

GitHub

2k stars
34 watching
255 forks
Language: Jupyter Notebook
last commit: 7 days ago
Linked from 3 awesome lists

jaxphysics-simulationreinforcement-learningrobotics

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
google-deepmind/mujoco A general-purpose physics simulator for simulating articulated structures in motion. 8,209
eleutherai/gpt-neox Provides a framework for training large-scale language models on GPUs with advanced features and optimizations. 6,941
avoroshilov/physics_playground A research framework for simulating coupled FEM and rigid body dynamics with real-time collision detection and optimization. 28
google-research/t5x A modular framework for training and deploying sequence models at scale 2,682
jrouwe/joltphysics A C++ library for simulating rigid body physics and collision detection in real-time applications like games and VR. 6,768
google-deepmind/dm_control A software stack for simulating physics-based environments and training reinforcement learning agents 3,814
bepu/bepuphysics2 A C# 3D physics simulation library with support for various rigid body shapes and constraints. 2,396
carperai/trlx A framework for distributed reinforcement learning of large language models with human feedback 4,502
sarmbruster/gatling-bolt Provides a Scala-based plugin to support Neo4j Bolt protocol in Gatling simulations 2
bulletphysics/bullet3 A C++ SDK for real-time collision detection and multi-physics simulation in various fields such as VR, games, and robotics. 12,653
alibaba/rax Builds applications that can run across multiple platforms and frameworks with minimal code changes. 7,995
nvidiagameworks/kaolin A PyTorch library for accelerating 3D deep learning research with various GPU-optimized operations and tools. 4,497
google-deepmind/mctx An open-source library providing efficient implementations of search algorithms for reinforcement learning 2,356
projectchrono/chrono A high-performance C++ library for simulating the dynamics of complex systems with rigid and flexible bodies, fluid-solid interactions, and other phenomena 2,244
google/trax An end-to-end deep learning library with clear code and speed 8,096