Sr Machine Learning Engineer- ML Infrastructure & Data Platforms

Senior Machine Learning Engineer – Applied Science Data Frameworks team. In this role, you’ll build the infrastructure powering large‑scale, multimodal AI training and inference. You’ll work across machine learning, distributed systems, and data engineering to develop tools and platforms that help teams train and deploy models at scale, supporting systems that process billions of data points across large GPU environments.

• Build distributed data loaders to support large‑scale training workflows
• Develop data pipelines for ingesting, transforming, and preparing multimodal datasets
• Design batch inference systems for high‑volume data processing across GPU environments
• Improve system performance, scalability, and reliability using distributed computing tools such as Ray, Spark, and DuckDB
• Implement search and retrieval systems using vector databases and embedding‑based approaches
• Develop and maintain CI/CD workflows, including testing, deployment, and containerization
• Partner with researchers and engineers to turn model requirements into scalable systems
• Create reusable tools, libraries, and documentation to support teams across the organization
• Monitor and improve system health, including throughput, latency, and resource utilization
• Support a collaborative team environment through code reviews and knowledge sharing

• 8+ years of experience building and operating distributed systems or ML infrastructure in production
• Experience working with large‑scale data pipelines or inference systems
• Strong programming skills in Python and a foundation in software engineering principles
• Experience with ML frameworks such as PyTorch or Tensor Flow
• Familiarity with distributed computing tools such as Ray, Spark, Dask, or similar
• Experience working with cloud platforms such as AWS or Azure
• Understanding of MLOps practices, including CI/CD and deployment workflows
• Ability to communicate clearly and collaborate with cross‑functional teams

• Master’s degree or Ph.D. in Computer Science, Machine Learning, or a related field (or equivalent practical experience)

• Experience working with multimodal data (images, video, text)
• Familiarity with vector databases or semantic search systems

Our compensation reflects the cost of labor across several U.S. geographic markets, and we pay differently based on those defined markets. The U.S. pay range for this position is $172,500–$306,625 annually. Pay within this range varies by work location and may also depend on job‑related knowledge, skills, and experience. In California, the pay range for this position is $211,800–$306,625. In Washington, the pay range for this position is $201,000–$291,150.