PyTorch
API: torch.distributed.ReduceOp.SUM
Strategy: Direct Mapping
JAX (Core)
API: —
Strategy: Macro ''sum''
NumPy
API: —
Strategy: Macro ''sum''
Keras
API: —
Strategy: Macro ''sum''
TensorFlow
API: —
Strategy: Macro ''sum''
Apple MLX
API: —
Strategy: Macro ''sum''
Flax NNX
API: —
Strategy: Macro ''sum''
PaxML / Praxis
API: —
Strategy: Macro ''sum''