ML Performance Engineer
TensorWave2 days ago
Las Vegas, NV, USAMid Level
Responsibilities
- Validate new cluster deployments by running large-scale ML training and inference workloads before customer handoff.
- Identify, reproduce, and diagnose performance and reliability issues with infrastructure and platform teams.
- Use managed Kubernetes and Slurm platforms from a customer workload perspective and provide improvement feedback.
- Run training and inference model benchmarks for internal and external audiences.
- Build tooling and repeatable workflows for cluster validation and benchmarking.
- Track distributed training and inference framework performance on AMD Instinct GPUs.
- Support customer clusters and workloads as part of the MLOps team.
Requirements
- 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 distributed training experience with 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.
- Ability to communicate performance findings clearly to engineering and non-technical audiences.
Benefits
- Stock options.
- 100% paid employee medical, dental, and vision insurance.
- Company health savings account contributions.
- 100% paid employee short-term and long-term disability insurance.
- Life and voluntary supplemental insurance options, including pet and legal insurance.
- Supplementary health benefits, including discounted virtual healthcare appointments and serious illness support.
- Flexible spending account and 401(k).
- Employee assistance program, flexible PTO, paid holidays, and parental leave.
- Other in-office perks.
- Employment is contingent upon verification of identity and authorization to work in the United States; background checks may apply where permitted by law.
Tech Stack
Categories
About TensorWave
TensorWave builds an AMD‑exclusive cloud platform for AI workloads, providing Instinct MI325X and MI355X GPU instances and tooling for training, fine‑tuning, and inference, including an inference engine. It sells infrastructure-as-a-service to startups and enterprises that need scalable AI compute. Founded in 2023 and headquartered in Las Vegas, Nevada, the company focuses on GPU‑accelerated cloud infrastructure for generative AI and machine learning teams.