Senior Software Engineer, Machine Learning Platform

Menlo Ventures; San Francisco (CA); Hybrid; USD 187,000 - 259,000; Full time

Benefits offered by this job

4 days in office per week

Fridays from home

Medical, dental, vision benefits

401k match

Generous vacation and paid days off

Parental leave

Job summary

Chime’s Machine Learning Platform (MLP) team builds the infrastructure, tooling, and developer experience that powers machine learning across the company. You will design and build scalable systems that support model training, feature computation, real-time inference, and experimentation, enabling data scientists and ML engineers to move quickly with reliability and cost awareness.

This role focuses on building robust foundations that allow ML teams to move quickly while maintaining governance

Qualifications

• 5+ years of experience in ML infrastructure, platform engineering, or production ML systems.
• Knowledge of the ML model development lifecycle including data preprocessing, training, evaluation and deployment.
• Experience with distributed systems, cloud computing, or large-scale data processing.
• Hands-on experience with CI/CD pipelines, DevOps practices, and infrastructure as code.

Responsibilities

• Design, build, and operate scalable ML infrastructure on AWS.
• Develop distributed training and batch processing systems using Ray.
• Build and maintain infrastructure-as-code using Terraform.
• Support and evolve the feature store and feature pipelines.
• Develop data ingestion and streaming systems (e.g., Kinesis, Kafka, Flink, Spark).
• Improve CI/CD workflows for ML models and platform components.
• Enhance observability, reliability, and cost visibility across ML workloads.
• Partner with Data Science and ML Engineering teams to improve developer experience.
• Contribute to platform architecture decisions and technical roadmaps.
• Participate in on-call rotations to support production systems.

Skills

Python

CI/CD

Distributed systems

Software engineering fundamentals

DevOps

Tools

Docker

Kubernetes

Terraform

Spark

Ray

AWS

Job description

About the role

Chime’s Machine Learning Platform (MLP) team builds and operates the infrastructure, tooling, and developer experience that powers machine learning across the company. We enable data scientists and ML engineers to develop, train, deploy, and monitor models reliably and efficiently.

As a Machine Learning Platform Engineer, you will design and build scalable systems that support model training, feature computation, real‑time inference, and experimentation. You’ll work at the intersection of distributed systems, cloud infrastructure, and applied machine learning.

This role focuses on building robust foundations that allow ML teams to move quickly while maintaining reliability, governance, and cost efficiency.

The base salary offered for this role and level of experience will begin at $187,000.00 and goes up to $259,000.00. Full‑time employees are also eligible for a bonus, competitive equity package, and benefits. The actual base salary offered may be higher, depending on your location, skills, qualifications, and experience.

In this role, you can expect to

• Design, build, and operate scalable ML infrastructure on AWS
• Develop distributed training and batch processing systems using Ray
• Build and maintain infrastructure-as-code using Terraform
• Support and evolve the feature store and feature pipelines
• Develop data ingestion and streaming systems (e.g., Kinesis, Kafka, Flink, Spark, or similar technologies)
• Improve CI/CD workflows for ML models and platform components
• Enhance observability, reliability, and cost visibility across ML workloads
• Partner closely with Data Science and ML Engineering teams to improve developer experience
• Contribute to platform architecture decisions and technical roadmaps
• Participate in on‑call rotations to support production systems

To thrive in this role, you have

• 5+ years of experience in ML infrastructure, platform engineering, or production ML systems
• Knowledge of the machine learning model development lifecycle, including data preprocessing, model training, evaluation, and deployment
• Experience with distributed systems, cloud computing, or large‑scale data processing
• Strong foundation in computer science and software engineering principles
• Deeply interested in the impact and evolution of advanced AI technologies
• Hands‑on experience with CI/CD pipelines, DevOps practices, and infrastructure as code
• Experience with containerization technologies such as Docker and Kubernetes, and orchestration systems
• Knowledge of cloud platforms such as AWS and distributed computing frameworks such as Spark and Ray
• Experience with GPU programming (CUDA) and GPU costs/optimization
• Strong programming skills in Python, Go, Scala, Java or similar languages
• Familiarity with infrastructure‑as‑code (e.g., Terraform, CloudFormation)
• Solid understanding of software engineering fundamentals (testing, version control, code review, observability)

Nice-to-have

• Experience with distributed compute frameworks such as Ray
• Experience building or operating a feature store
• Experience with real‑time ML systems or model serving
• Familiarity with streaming technologies (Kafka, Kinesis, Flink, Spark Streaming, etc.)
• Experience supporting ML lifecycle workflows (training, evaluation, deployment, monitoring)
• Knowledge of ML experimentation platforms and model governance practices

What we offer for our full‑time, regular employees

• Our in‑office work policy is designed to keep you connected - with four days a week in the office and Fridays from home for those near one of our offices, plus team and company‑wide events depending on location. Whether you’re coming in regularly or are part of our fully remote program, you’ll stay engaged with your work and teammates.
• In‑office perks including backup child, elder, and/or pet care, plus a subsidized commuter benefit to support your regular commute
• Competitive salary based on experience
• 401k match plus great medical, dental, vision, life, and disability benefits
• Generous vacation policy and company‑wide Chime Days, bonus company‑wide paid days off
• 1% of your time off to support local community organizations of your choice
• Annual wellness stipend to use towards eligible wellness related expenses
• Up to 24 weeks of paid parental leave for birthing parents and 12 weeks of paid parental leave for non‑birthing parents
• Access to Maven, a family planning tool, with $15k lifetime reimbursement for egg freezing, fertility treatments, adoption, and more.
• In‑person and virtual events to connect with your fellow Chimers—think cooking classes, guided meditations, music festivals, mixology classes, paint nights, etc., and delicious snack boxes, too!
• A challenging and fulfilling opportunity to join one of the most experienced teams in FinTech and help millions unlock financial progress

Chime is proud to be an Equal Opportunity Employer. We consider qualified applicants without regard to race, color, ancestry, religion, sex, national origin, sexual orientation, gender identity, age, marital or family status, disability, genetic information, veteran status, or any other legally protected basis under provincial, federal, state, and local laws, regulations, or ordinances. We will also consider qualified applicants with criminal histories in a manner consistent with the requirements of state and local laws, including the San Francisco Fair Chance Ordinance, Cook County Ordinance, NYC Fair Chance Act, and the LA City Fair Chance Ordinance, and consistent with Canadian provincial and federal laws. If you have a disability or special need that requires accommodation during any stage of the application process, please contact: [e-mail].