Machine Learning Infrastructure Engineer

Requirements:

• 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

Nice to have:

• Experience with large GPU clusters and high-performance computing/networking
• Experience with supporting large language model training
• Experience with ML frameworks like Pytorch/TensorFlow/JAX
• Experience with GPU kernel development