PlusAI
Senior Machine Learning Infrastructure Engineer
$160K–$200K
Santa Clara, CA · Hybrid · Senior · Full-time · Posted Jul 15, 2024
- Experience
- PhD new graduates or master’s graduates with 3+ years of software engineering experience focused on ML infrastructure or distributed systems.
- Education
- PhD or MS in Computer Science, Electrical Engineering, or a related field.
About the job
Designs and operates scalable machine-learning infrastructure for training, inference, deployment, and monitoring, including data pipelines, GPU clusters, and model reproducibility systems.
What you'll do
- Build and maintain scalable ML systems, data pipelines, model versioning, and experiment tracking.
- Improve distributed storage and CI/CD workflows, and ensure platform reliability through monitoring, logging, and alerting.
- Collaborate with ML teams, mentor junior engineers, and support quality and process improvements.
What they ask for
- PhD or MS in Computer Science, Electrical Engineering, or a related field.
- PhD new graduates or master’s graduates with 3+ years of software engineering experience focused on ML infrastructure or distributed systems.
- Proficiency in Python, C++, and SQL.
- Experience with cloud or on-prem resource management, distributed systems, and at least one deep learning framework.
Nice to have
- Familiarity with CNNs or Transformer models; experience building large-scale ML datasets, MLOps pipelines, or distributed computing frameworks such as Ray.
- Experience with autonomous vehicles or robotics.
Pay
$160,000 - $200,000 a year (base salary)
Tools
Python, C++, SQL, Docker, Kubernetes, Kubeflow, mlflow, PyTorch
AI focus
Builds infrastructure for machine-learning training, inference, deployment, monitoring, and reproducibility.
Model-written summary · the original posting is authoritative