date = 2022-01-23
tags = ["python", "jax"]
title = "JAX API"

JAX API

Google's JAX autograd library

is an example of a neat, composable API.

This notebook

shows all that is needed to plot the gradient field of a function using JAX's `grad` and `vmap` functions.

Comments

Leave a comment