ML Performance Engineer

• Validate new cluster deployments by running large-scale training and inference workloads before customer handoff
• Identify, reproduce, and diagnose performance and reliability issues, working with infrastructure and platform teams to resolve them
• Use our managed Kubernetes and Slurm platforms the way our customers do, and provide feedback that improves them
• Run model benchmarks across training and inference, both for internal and external use
• Build tooling and repeatable workflows for validation and benchmarking
• Stay current with distributed training and inference frameworks and their performance on AMD Instinct GPUs
• Work with larger MLOps team to help support customer clusters and workloads

• Bachelor of Science in Computer Science, Computer Engineering, or a related technical field, or equivalent practical experience
• 3+ years of experience running ML workloads on GPU clusters
• Hands-on experience with distributed training using frameworks such as PyTorch FSDP, Megatron-LM, DeepSpeed, or torchtitan
• Working knowledge of Slurm and Kubernetes from a user and workload perspective
• Experience profiling and optimizing GPU workloads, including collective communication performance
• Strong Python skills and solid Linux fundamentals
• Clear written communication of performance findings to engineering and non-technical audiences

By submitting an application, you acknowledge that TensorWave may collect, use, and retain your personal information for recruiting and employment-related purposes in accordance with applicable data privacy laws.

• Stock Options
• 100% paid Medical, Dental, and Vision insurance for Employees
• Company Health Savings Account Contributions
• 100% paid Short Term and Long Term Disability Insurance for Employees
• Life and Voluntary Supplemental Insurance Options
• Other Insurance Options, such as Pet & Legal Insurance
• Various Supplementary Health Benefits, such as discounted Virtual Healthcare Appointments and Serious Illness Support
• Flexible Spending Account
• 401(k)
• Employee Assistance Program
• Flexible PTO
• Paid Holidays
• Parental Leave
• Other In-Office Perks

Our mission is simple: deliver seamless, secure, reliable, and resilient AI compute at scale. We've built a versatile cloud platform that eliminates infrastructure barriers, empowering builders to focus on innovation instead of fighting their stack. Because breakthrough AI should move at the speed of ideas, not infrastructure.