Machine Learning Infrastructure Engineer

Job summary

Responsibilities

We’re looking for seasoned ML Infrastructure engineers with experience designing, building and maintaining training and serving infrastructure for ML research.

• Provide infrastructure support to our ML research and product
• Build tooling to diagnose cluster issues and hardware failures
• Monitor deployments, manage experiments, and generally support our research
• Maximize GPU allocation and utilization for both serving and training

What they're looking for

• 4+ years of experience supporting the infrastructure within an ML environment
• Experience in developing tools used to diagnose ML infrastructure problems and failures
• Experience with cloud platforms (e.g., Compute Engine, Kubernetes, Cloud Storage)
• Experience working with GPUs

Tech stack

• Kubernetes
• PyTorch
• TensorFlow
• JAX