Our great sponsors
-
tiny-differentiable-simulator
Tiny Differentiable Simulator is a header-only C++ and CUDA physics library for reinforcement learning and robotics with zero dependencies.
-
WorkOS
The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
I need differentiable rigid body dynamics because I want to do nonlinear MPC. One library that can do this is C++ is Tiny Differentiable Simulator https://github.com/erwincoumans/tiny-differentiable-simulator. As I understand it, this software uses a C++ auto-diff library and code generation to create CUDA kernels to compute fast derivatives in parallel. This seems pretty fast because it's C++. Another option is Brax https://github.com/google/brax. Brax uses JAX which I've never used, but from what I've seen online, JAX is popular for researchers and probably very good.
I need differentiable rigid body dynamics because I want to do nonlinear MPC. One library that can do this is C++ is Tiny Differentiable Simulator https://github.com/erwincoumans/tiny-differentiable-simulator. As I understand it, this software uses a C++ auto-diff library and code generation to create CUDA kernels to compute fast derivatives in parallel. This seems pretty fast because it's C++. Another option is Brax https://github.com/google/brax. Brax uses JAX which I've never used, but from what I've seen online, JAX is popular for researchers and probably very good.
Related posts
- 4000x Speedup in Reinforcement Learning with Jax
- Physic engine for 3D simulation: which one to use?
- Best environment to train RL agents
- [R] Brax: A Differentiable Physics Engine for Large Scale Rigid Body Simulation, with a focus on performance and parallelism on accelerators, written in JAX.
- Brax – A Differentiable Physics Engine for Large Scale Rigid Body Simulation