Provides an implementation of a missing primitive in JAX, value_and_jacfwd - View it on GitHub