This repository contains code examples used in the Medium article:

“High-order derivatives using JAX Taylor-mode automatic differentiation”

🔗 Read the article on Medium »

• example_nested_vs_taylor.ipynb Compares performance of Taylor-mode automatic differentiation using JAX's jet module against nested applications of jax.grad.
• example_sin.ipynb Demonstrates how to compute the Taylor series expansion of simple functions using the jet module.
• example_multivariate.ipynb Shows how to use jet for multivariate differentiation.

• robochimps/vibrojet