PyTorch
API: torch.align_tensors
Strategy: Direct Mapping
JAX (Core)
API: jnp.broadcast_arrays
Strategy: Direct Mapping
NumPy
API: np.broadcast_arrays
Strategy: Direct Mapping
Keras
API: keras.ops.broadcast_tensors
Strategy: Direct Mapping
TensorFlow
API: —
Strategy: Custom / Partial
Apple MLX
API: mx.broadcast_arrays
Strategy: Direct Mapping
Flax NNX
API: jnp.broadcast_arrays
Strategy: Direct Mapping
PaxML / Praxis
API: jnp.broadcast_arrays
Strategy: Direct Mapping