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.

语言：Jupyter Notebook · GitHub 星标：185 · Fork：8

话题：batch · casadi · gpu · jax

在 GitHub 上查看原仓库