Links to a bunch of Jupyter/Colab notebooks that show my steps on
- Learning to use jax and creating a simple mass-spring system
- Speeding up the physics code using
scan
andjit
- Using differentiable physics to optimize the mass-spring system's physics properties
- Using differentiable physics to optimize a mass-spring system's physics properties and even start-pose