Add has_aux to jacrev, jacfwd and hessian

Ask questionsAdd has_aux to jacrev, jacfwd and hessian

Allow functions that return aux information, similarly to grad or jvp.

google/jax

Answer questions sguada

Duplicate of https://github.com/google/jax/issues/497

useful!

Related questions

Installation problem hot 1

Installing from source using Conda and CUDA could be improved - jax hot 1

jax `odeint` fails against scipy `odeint` hot 1

cuda failed to allocate errors hot 1

Custom VJPs for external functions hot 1

cuda failed to allocate errors hot 1

Unimplemented NumPy core functions hot 1

Reshape layer for stax - jax hot 1

Installing from source using Conda and CUDA could be improved hot 1

Clear GPU memory hot 1

jax/stax BatchNorm: running average on the training set and l2 regularisation hot 1