adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.
BSD 3-Clause "New" or "Revised" License
131
stars
20
forks
source link
Update README.md for Jax usage #91
Closed
Giulero closed 4 months ago
A small update of the readme. Adding some details on jit and vmap of jax functions.