PyTorch
API: torch.uint8
Strategy: Plugin (type_cast_shim)
JAX (Core)
API: jax.numpy.uint8
Strategy: Direct Mapping
NumPy
API: numpy.ubyte
Strategy: Direct Mapping
Keras
API: —
Strategy: Custom / Partial
TensorFlow
API: tf.uint8
Strategy: Direct Mapping
Apple MLX
API: mlx.core.uint8
Strategy: Direct Mapping
Flax NNX
API: jax.numpy.uint8
Strategy: Direct Mapping
PaxML / Praxis
API: jax.numpy.uint8
Strategy: Direct Mapping