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.
[x] Pytorch from HLO (optional module) currently impossible
[x] Compare pytorch (cusadi), pytorch (our), jax, cuda, cpu, mjx We compare cusadi(cuda) vs jaxadi directly. Comparison of pinocchio+casadi+jaxadi vs mjx is yet to come
Pytorch from HLO (optional module)currently impossibleCompare pytorch (cusadi), pytorch (our), jax, cuda, cpu, mjxWe compare cusadi(cuda) vs jaxadi directly. Comparison of pinocchio+casadi+jaxadi vs mjx is yet to comePytrees + comparison(gave up the idea)