Software Engineer, SystemML - AI Networking

Highlights

• Own the communication stack behind nearly every distributed GPU workload.
• Improve reliability and performance for large-scale GenAI and LLM training.
• Lead collective communication library development for GPU training infra.

What they offer

• bonus
• equity
• benefits

Ideal candidate

A systems-minded engineer who leads cross-functional projects and cares about reliable, high-performance distributed AI at scale, with passion for GenAI/LLM challenges.

Requirements

• Proven C/C++ and Python programming skills
• Proven track record of leading successful projects
• Experience leading cross-functional technical projects and communicating technical decisions
• Specialized experience in distributed ML training, GPU architecture, ML systems, AI infrastructure, HPC, performance optimization, or ML frameworks
• Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience