tf.keras.config.is_nnx_enabled

Checks whether NNX specific features are enabled for the JAX backend.

bool True if NNX backend features are enabled, False otherwise. Defaults to False.