ML Performance Engineer
Confirmed on the employer's own hiring board on Nov 14, 2026. First seen by Alion on Nov 13, 2026. TensorWave scores B on the Alion truth index.
About TensorWave
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.
About the Role
We're looking for an ML Performance Engineer to join our MLOps team. You will validate new clusters by running large-scale training and inference workloads, help improve our managed Kubernetes and Slurm platforms through hands-on use, and produce performance benchmarks for customers and the broader market.
What You’ll Do
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
Who You Are
Required Qualifications
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
What We Offer
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
Equal Employment Opportunity
TensorWave is an Equal Opportunity Employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. We do not discriminate on the basis of any protected status under applicable law.
Reasonable Accommodations
TensorWave provides reasonable accommodations in accordance with applicable laws. If you require accommodation during the hiring process, please contact [email protected].
Employment Eligibility
All offers of employment are contingent upon verification of identity and authorization to work in the United States, as required by law.
Background Checks
Where permitted by law, employment may be contingent upon the successful completion of a job-related background check.
Data Privacy Notice
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.