Software Engineer, Machine Learning Platform

Chime

Hybrid Actively hiring Posted this week
San Francisco, CA Posted 3 days ago $259,000–$259,000 / year
View original post Log in to save

At a glance

AI generated

TL;DR

As a Machine Learning Platform Engineer at Chime’s MLP team, you will design and build scalable systems for model training, feature computation, real-time inference, and experimentation on AWS. You’ll develop distributed training and batch processing systems using Ray, maintain infrastructure-as-code with Terraform, and support CI/CD workflows while enhancing observability and reliability across ML workloads. This role requires 5+ years of experience in ML infrastructure or production ML systems, expertise in cloud computing, distributed systems, and hands-on skills with Python, Go, Scala, Java, Docker, Kubernetes, and AWS services like Spark and Ray. You will collaborate closely with Data Science and ML Engineering teams to improve developer experience and contribute to platform architecture decisions, ensuring robust foundations for rapid yet reliable ML development at scale.

Skills

AWS Terraform Ray Docker Kubernetes CI/CD Prometheus Grafana Python Go Spark CUDA Kafka Kinesis Flink

What you'll do

  • Design and build scalable machine learning infrastructure on AWS.
  • Develop distributed training systems using Ray for efficient model training.
  • Maintain infrastructure-as-code using Terraform to ensure reliability and scalability.
  • Support and enhance the feature store and feature pipelines for data scientists.
  • Improve observability, reliability, and cost efficiency across ML workloads.

What we're looking for

  • 5+ years experience in ML infrastructure or production ML systems
  • Expertise in distributed systems, cloud computing, and large-scale data processing
  • Hands-on experience with CI/CD pipelines, DevOps practices, and infrastructure as code
  • Proficiency in Python, Go, Scala, Java, Docker, Kubernetes, and AWS
  • Knowledge of machine learning model lifecycle including training, evaluation, deployment
  • Experience with GPU programming and optimization for cost efficiency

Employer

About Chime

Chime is a financial technology company offering mobile-first banking services including fee-free checking accounts, savings accounts, and a secured credit builder card through partner banks. Industry: Financial Technology & Neobanking

Chime currently has 25 open roles on FindRole.

Listed pay typically runs $227,000–$227,000 across 25 roles with salary data.

Most-posted roles

View all roles at Chime