Sr Machine Learning Engineer- ML Infrastructure & Data Platforms

Job description

We’re looking for a Senior Machine Learning Engineer to join our Applied Science Data Frameworks team. In this role, you’ll build the infrastructure that powers 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. Your work will support systems that process billions of data points across large GPU environments. If you’re motivated by solving complex problems and building systems that enable others to do their best work, we’d love to connect. What You’ll Do  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 (e.g., Ray, Spark, 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,…