issues
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
feat: ahead of time jax function compilation
#1
Closed
lvjonok
closed
2 weeks ago