BFloat16

Data type constant for brain floating point 16.

PyTorch

API: torch.bfloat16
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.bfloat16
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.bfloat16
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.bfloat16
Strategy: Direct Mapping
Official Docs ↗