Lead Machine Learning Engineer

Capital One Capital One · Banking · San Francisco, CA

Lead Machine Learning Engineer role focused on productionizing ML applications and systems at scale within a fintech company. Responsibilities include designing, building, and delivering ML models and components, informing ML infrastructure decisions, writing and testing application code, automating tests and deployment, retraining and monitoring models in production, leveraging cloud architectures, constructing data pipelines, and ensuring code quality, model governance, and responsible AI practices. Requires experience in designing data-intensive solutions, programming, and building/scaling ML systems.

What you'd actually do

  1. Design, build, and/or deliver ML models and components that solve real-world business problems, while working in collaboration with the Product and Data Science teams
  2. Inform your ML infrastructure decisions using your understanding of ML modeling techniques and issues, including choice of model, data, and feature selection, model training, hyperparameter tuning, dimensionality, bias/variance, and validation)
  3. Solve complex problems by writing and testing application code, developing and validating ML models, and automating tests and deployment
  4. Retrain, maintain, and monitor models in production
  5. Leverage or build cloud-based architectures, technologies, and/or platforms to deliver optimized ML models at scale.

Skills

Required

  • Bachelor's Degree
  • 6 years of experience designing and building data-intensive solutions using distributed computing
  • 4 years of experience programming with Python, Scala, or Java
  • 2 years of experience building, scaling, and optimizing ML systems

Nice to have

  • Master's or Doctoral Degree in computer science, electrical engineering, mathematics, or a similar field
  • 3+ years of experience building production-ready data pipelines that feed ML models
  • 3+ years of on-the-job experience with an industry recognized ML framework such as scikit-learn, PyTorch, Dask, Spark, or TensorFlow
  • 2+ years of experience developing performant, resilient, and maintainable code
  • 2+ years of experience with data gathering and preparation for ML models
  • 2+ years of people leader experience
  • 1+ years of experience leading teams developing ML solutions using industry best practices, patterns, and automation
  • Experience developing and deploying ML solutions in a public cloud such as AWS, Azure, or Google Cloud Platform
  • Experience designing, implementing, and scaling complex data pipelines for ML models and evaluating their performance
  • ML industry impact through conference presentations, papers, blog posts, open source contributions, or patents
  • Experience leveraging interactive AI tooling to accelerate productivity, utilizing capabilities beyond basic code completion

What the JD emphasized

  • productionizing machine learning applications and systems at scale
  • high availability and performance
  • solve complex problems
  • automating tests and deployment
  • optimized ML models at scale
  • optimized data pipelines
  • test automation
  • Responsible and Explainable AI

Other signals

  • productionizing machine learning applications and systems at scale
  • develop and review model and application code
  • ensure high availability and performance of our machine learning applications
  • design, build, and/or deliver ML models and components
  • retrain, maintain, and monitor models in production
  • construct optimized data pipelines to feed ML models
  • leverage continuous integration and continuous deployment best practices
  • ensure all code is well-managed to reduce vulnerabilities
  • models are well-governed from a risk perspective
  • ML follows best practices in Responsible and Explainable AI