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
jetmodule against nested applications ofjax.grad. - example_sin.ipynb
Demonstrates how to compute the Taylor series expansion of simple functions using the
jetmodule. - example_multivariate.ipynb
Shows how to use
jetfor multivariate differentiation.