AI Training Frameworks
PyTorch and TensorFlow, tuned for the interconnect
Extensions and optimizations for PyTorch and TensorFlow. Includes custom NCCL/RCCL communicators for multi-GPU all-reduce, ZeRO-3 partitioning (model states, gradients, and optimizers), mixed-precision trainers (FP8/BF16), asynchronous checkpointing, and elastic training that dynamically adds/removes nodes mid-run.
- PyTorch
- TensorFlow
- NCCL
- RCCL
- ZeRO-3
- FP8
- BF16
