Senior ML Systems Engineer — Research Infrastructure

EPM Scientific; California; On-site; USD 180,000 - 260,000; Full time

Job summary

EPM Scientific is seeking a Senior / Staff Machine Learning Research Engineer to lead the design of scalable ML infrastructure in California. The role sits at the intersection of research and systems engineering, owning platforms for distributed training, experimentation, and model deployment.

You will collaborate with researchers to turn novel architectures into production-grade implementations, optimize training and inference, and improve developer experience across a growing ML organization.

Qualifications

• MS or PhD in Computer Science, ML, Engineering, or a related technical discipline.
• Proven experience building machine learning systems that were widely adopted by researchers, engineers, or end users.
• Deep hands-on experience with both PyTorch and JAX.
• Experience scaling machine learning workloads using distributed training techniques across GPU and/or TPU environments.
• Strong understanding of model optimization, accelerator performance, memory management, training throughput, and parallelization strategies.
• Excellent software engineering fundamentals, including system design, API design, testing, maintainability, and developer tooling.
• Demonstrated ability to independently drive ambiguous, multi-quarter technical initiatives from concept through production.
• Comfort operating in a fast-moving environment where priorities evolve alongside scientific and technical discovery.

Responsibilities

• Design and build the core platform that supports large-scale machine learning development, including distributed training orchestration, asynchronous execution, experiment management, and high-throughput data pipelines.
• Partner with researchers to transform novel model architectures into efficient, production-grade implementations, optimizing training and inference workloads and solving bottlenecks across large-scale compute environments.
• Build and maintain tooling, libraries, and workflows that support model development in PyTorch and JAX, improving reproducibility, scalability, experimentation velocity, and developer experience.
• Establish best practices around testing, reproducibility, model versioning, evaluation, code quality, and experiment tracking to improve reliability without slowing research velocity.
• Collaborate with researchers and data engineers to translate cutting-edge ideas into practical systems, bridging research and engineering.

Skills

PyTorch

JAX

Distributed training

System design

API design

Experimentation tooling

Education

MS or PhD in Computer Science / ML / Engineering

Job description

EPM Scientific is seeking a Senior / Staff Machine Learning Research Engineer to lead the design of scalable ML infrastructure in California. The role sits at the intersection of research and systems engineering, owning platforms for distributed training, experimentation, and model deployment.

You will collaborate with researchers to turn novel architectures into production-grade implementations, optimize training and inference, and improve developer experience across a growing ML organization.