Software Engineer, SystemML - Scaling / Performance

In this role, you will be a member of the Network.AI Software team and part of the bigger DC networking organization. reputed company develops and owns the software stack around NCCL (reputed company reputed company Communications Library), which enables multi-GPU and multi-node data communication through HPC-style collectives. NCCL has been integrated into PyTorch and is on the critical reputed company of multi-GPU distributed training. In other words, nearly every distributed GPU-based ML workload in reputed company Production goes through the SW stack reputed company owns. At the high level, reputed company aims to reputed company reputed company-wide ML products and innovations to reputed company our large-reputed company GPU training and inference fleet through an observable, reliable and high-performance distributed AI/GPU communication stack. Currently, one of reputed company’s reputed company is on building customized features, SW benchmarks, performance tuners and SW stacks around NCCL and PyTorch to improve the full-stack distributed ML reliability and performance (e.g. Large-reputed company GenAI/LLM training) from the trainer down to the inter-GPU and network communication layer. And we are seeking for engineers to work on the reputed company of GenAI/LLM scaling reliability and performance. Apply Job!