DistributedSum

Enum value for Sum reduction in distributed operations.

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.distributed.ReduceOp.SUM

Strategy: Direct Mapping

Official Docs ↗

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''