Machine Learning Infrastructure Engineer
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
The role
The role focuses on designing, building, and maintaining scalable ML training and serving infrastructure to accelerate research and product development.
You will develop tooling to diagnose cluster issues and hardware failures, monitor deployments, and manage experiments.
A core priority is maximizing GPU allocation and utilization for both training and serving workloads.
Provide infrastructure support to ML teams, troubleshoot performance bottlenecks, and coordinate with platform engineers.
Required hands-on experience with cloud platforms, Kubernetes, and GPU-enabled environments, plus experience with PyTorch, TensorFlow, or JAX.
This role demands 4+ years of ML infra experience and a proactive, collaborative mindset in a fast-paced research setting.
You will develop tooling to diagnose cluster issues and hardware failures, monitor deployments, and manage experiments.
A core priority is maximizing GPU allocation and utilization for both training and serving workloads.
Provide infrastructure support to ML teams, troubleshoot performance bottlenecks, and coordinate with platform engineers.
Required hands-on experience with cloud platforms, Kubernetes, and GPU-enabled environments, plus experience with PyTorch, TensorFlow, or JAX.
This role demands 4+ years of ML infra experience and a proactive, collaborative mindset in a fast-paced research setting.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
or
Already have an account?
Log inSimilar openings
Other roles that could suit you.
? Machine Learning Intern Fall 2026 (Toronto) ? Machine Learning Engineer, Behavior Internship ? Machine Learning Engineer (L3) ? Senior C++ Programmer - Machine Learning Content Creation Technology Group ? Senior C++ Programmer - Machine Learning - Content Creation Technology Group ? Team Lead - Machine Learning Engineer
Remote workno
CityRedwood City, United States