search

based-robotics / jaxadi

Transforms your CasADi functions into batchable JAX-compatible functions. By combining the power of CasADi with the flexibility of JAX, JAXADi enables the creation of efficient code that runs smoothly on CPUs, GPUs, and TPUs.
https://based-robotics.github.io/jaxadi/
MIT License
94 stars 2 forks source link

MJX comparison #2

Closed simeon-ned closed 1 week ago

simeon-ned commented 2 weeks ago

The comparison to Mujoco XLA, maybe beneficial to ensure that generating stand alone functions is better the just using MJX