set_autodiff_mode

static Config.set_autodiff_mode(enable)

Enable or disable automatic differentiation mode.

Note

This setting controls how the role-specific proxies are resolved: unp_static remains backed by numpy. When autodiff mode is enabled, unp_dynamic and unp_analysis use jax.numpy. When autodiff mode is disabled, unp_analysis falls back to numpy, while unp_dynamic follows the selected execution backend and may still use jax.numpy on GPU.

Parameters:

enable (bool) – True or False.