PyTorch
API: torch.get_default_dtype
Strategy: Direct Mapping
JAX (Core)
API: —
Strategy: Macro 'jnp.float32'
NumPy
API: —
Strategy: Macro 'np.float64'
Keras
API: keras.backend.floatx
Strategy: Direct Mapping
TensorFlow
API: tf.keras.backend.floatx
Strategy: Direct Mapping
Flax NNX
API: —
Strategy: Macro 'jnp.float32'
PaxML / Praxis
API: —
Strategy: Macro 'jnp.float32'