💾 Archived View for jaeyoung.se › posts › jax-api captured on 2023-07-10 at 13:20:23. Gemini links have been rewritten to link to archived content
⬅️ Previous capture (2023-03-20)
-=-=-=-=-=-=-
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.