set_autodiff_mode

static Config.set_autodiff_mode(enable)

Enable or disable automatic differentiation mode.

Note

If True, sets unp to jax.numpy. If False, sets unp to numpy.

Parameters:

enable (bool) – True or False.