Senior Machine Learning Engineer
TensorWave11 months ago
Las Vegas, NV, USASenior
Responsibilities
- Design, operate, and improve ML infrastructure systems supporting distributed training and inference workloads.
- Build reliable and repeatable workload execution and orchestration patterns across shared GPU environments.
- Troubleshoot performance, reliability, and scalability issues across the ML stack.
- Partner with ML, systems, and platform teams to improve developer experience and operational efficiency.
Requirements
- Bachelor of Science in Computer Science, Computer Engineering, or a related technical field, or equivalent practical experience.
- Expertise supporting production machine learning systems using SLURM and Kubernetes.
- Strong understanding of GPU-accelerated workloads and distributed systems concepts.
- Solid Linux fundamentals and experience debugging infrastructure-level issues.
- Ability to build automation and tooling with Python, Go, or similar languages.
- Experience with schedulers, orchestration platforms, or cluster managers is preferred.
- Familiarity with large-scale GPU environments or HPC-style systems is preferred.
- Experience improving infrastructure reliability, utilization, or performance at scale is preferred.
Benefits
- Stock options
- 100% paid medical, dental, and vision insurance
- Flexible PTO
- Paid holidays
- 401(k)
- Parental leave
- Flexible Spending Account
- Short-term disability insurance
- Life and voluntary supplemental insurance
- Mental health benefits through Spring Health
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.