Senior ML Engineer - Platform & Production

Capital One Group; McLean (VA); On-site; USD 197,000 - 225,000; Full time

Job summary

Capital One in McLean, VA is seeking a Machine Learning Engineer to join the Enterprise ML Platform team. You will help build a high-throughput, multi-tenant training platform powering model workflows across Capital One, from classical ML to GenAI, using Ray, Spark, Dask, and Kubeflow Pipelines v2 on Amazon EKS.

You’ll work with product and data science colleagues to design, train, and deploy ML models, automate tests and deployment, and ensure governance, security, and responsible AI practices

Qualifications

• Bachelor's degree in Computer Science or related quantitative field.
• 4+ years of experience programming with Python, Java, Golang, or C++.
• 4+ years of ML experience using PyTorch or TensorFlow.
• 4+ years deploying and operating ML in production and cloud environments (AWS/GCP/Azure) with Kubernetes.

Responsibilities

• Design, build, and deliver ML models and components that solve real-world business problems, collaborating with Product and Data Science teams.
• Inform ML infrastructure decisions using modeling techniques, data, feature selection, and validation considerations.
• Solve complex problems by writing and testing code, developing and validating ML models, and automating tests and deployment.
• Collaborate in a cross-functional Agile team to create software enabling state-of-the-art big data and ML applications.
• Retrain, maintain, and monitor models in production.
• Leverage cloud architectures and platforms to deliver optimized ML models at scale.
• Construct optimized data pipelines to feed ML models.
• Apply CI/CD, test automation, monitoring for deployment; ensure secure, governed, and responsible AI practices.
• Use programming languages like Python, Scala, or Java.

Skills

Python

Java

Golang

C++

PyTorch

TensorFlow

Spark

Ray

Kubernetes

AWS

GCP

Azure

Education

Bachelor's Degree in CS or related field

Tools

Kubeflow

Docker

Job description

Capital One in McLean, VA is seeking a Machine Learning Engineer to join the Enterprise ML Platform team. You will help build a high-throughput, multi-tenant training platform powering model workflows across Capital One, from classical ML to GenAI, using Ray, Spark, Dask, and Kubeflow Pipelines v2 on Amazon EKS.

You’ll work with product and data science colleagues to design, train, and deploy ML models, automate tests and deployment, and ensure governance, security, and responsible AI practices