Repeated application of grad?

[cgiovanetti]

Hi all, sometime at our meeting last month I think someone was looking for a built-in function that repeatedly applied grad. What was the resolution of this?

[nstarman]

https://jax.readthedocs.io/en/latest/jax.experimental.jet.html