CastBFloat16 ============ Casts input to bfloat16. **Abstract Signature:** ``CastBFloat16(x)`` .. raw:: html

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