jax.jacobian

jax.jacobian(fun, argnums=0, has_aux=False, holomorphic=False, allow_int=False)

Alias of jax.jacrev().

Parameters:

• fun (Callable)

• argnums (int | Sequence[int])

• has_aux (bool)

• holomorphic (bool)

• allow_int (bool)

Return type:

Callable