PyTorch
API: torch.bfloat16
Strategy: Infix (?)
JAX (Core)
API: jnp.bfloat16
Strategy: Direct Mapping
NumPy
API: —
Strategy: Custom / Partial
Keras
API: —
Strategy: Custom / Partial
TensorFlow
API: tf.bfloat16
Strategy: Direct Mapping
Apple MLX
API: mlx.core.bfloat16
Strategy: Direct Mapping
Flax NNX
API: jnp.bfloat16
Strategy: Direct Mapping
PaxML / Praxis
API: jnp.bfloat16
Strategy: Direct Mapping