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
115 stars 3 forks source link

Colab Demo Notebook for JaxADi #11

Closed simeon-ned closed 1 month ago

simeon-ned commented 1 month ago

I would like to suggest a Google Colab demo for jaxadi, which transforms CasADi functions into batchable JAX-compatible functions. This will showcase:

  1. Setup and installation.
  2. Step-by-step usage of jaxadi.
  3. Example inputs and outputs.
  4. Key features and customization.

Benefits: