💾 Archived View for jaeyoung.se › posts › jax-api captured on 2023-03-20 at 17:44:27. Gemini links have been rewritten to link to archived content
-=-=-=-=-=-=-
date = 2022-01-23 tags = ["python", "jax"] title = "JAX API"
is an example of a neat, composable API.
shows all that is needed to plot the gradient field of a function using JAX's `grad` and `vmap` functions.