profile
viewpoint

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
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
Github User Rank List