Senior Machine Learning Engineer

About the role

TensorWave is seeking a Senior Machine Learning Engineer to design, build, and operate the core infrastructure that powers large‑scale training and inference on its GPU platform. You will collaborate with cross‑functional teams to ensure the platform meets performance, reliability, and scalability goals while enabling developers to focus on innovation.

Key responsibilities

• Design, deploy, and continuously improve ML infrastructure for distributed training and inference workloads.
• Create reliable, repeatable execution and orchestration patterns across shared GPU environments.
• Diagnose and resolve performance, reliability, and scalability issues throughout the ML stack.
• Partner with ML, systems, and platform teams to enhance developer experience and operational efficiency.

Required profile

• Bachelor’s degree in Computer Science, Computer Engineering, or a related technical field, or equivalent practical experience.
• Proven experience supporting production ML systems using SLURM and Kubernetes.
• Strong understanding of GPU‑accelerated workloads and distributed systems concepts.
• Solid Linux fundamentals and ability to debug infrastructure‑level issues.
• Ability to build automation and tooling using languages such as Python and Go.

Required skills

• SLURM
• Kubernetes
• GPU‑accelerated workloads
• Distributed systems
• Linux
• Python
• Go

What we offer

• Stock options
• 100% paid medical, dental, and vision insurance
• Health Savings Account contributions
• Short‑ and long‑term disability insurance
• Life and supplemental insurance options
• Flexible Spending Account and 401(k) plan
• Flexible PTO, paid holidays, and parental leave
• In‑office perks and additional health benefits