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
33 watching
259 forks
Language: Jupyter Notebook
last commit: about 1 month 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,338
eleutherai/gpt-neox Provides a framework for training large-scale language models on GPUs with advanced features and optimizations. 6,997
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,706
jrouwe/joltphysics A C++ library for simulating rigid body physics and collision detection in real-time applications like games and VR. 6,950
google-deepmind/dm_control A software stack for simulating physics-based environments and training reinforcement learning agents 3,849
bepu/bepuphysics2 A C# 3D physics simulation library with support for various rigid body shapes and constraints. 2,417
carperai/trlx A framework for distributed reinforcement learning of large language models with human feedback 4,537
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,777
alibaba/rax Builds applications that can run across multiple platforms and frameworks with minimal code changes. 7,994
nvidiagameworks/kaolin A PyTorch library for accelerating 3D deep learning research with various GPU-optimized operations and tools. 4,550
google-deepmind/mctx An open-source library providing efficient implementations of search algorithms for reinforcement learning 2,377
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,266
google/trax An end-to-end deep learning library with clear code and speed 8,114