Machine Learning Engineer

Job Summary

A company is looking for a Software Engineer (ML Infra) with extensive machine learning infrastructure and operations experience.

Key Responsibilities

• Collaborate with Data Scientists to design and implement model training and evaluation workflows
• Optimize machine learning models built with JAX and TensorFlow
• Manage data processing from ingestion to model inference and build tools for data analysts

Required Qualifications

• Deep experience optimizing ML systems for GPUs and CPUs
• Proficient in JAX or similar tools
• Experience working with Data Scientists on complex model training workflows
• Familiarity with DevOps tools such as Docker, Kubernetes, and Terraform
• Understanding of machine learning and statistical techniques