Skip to content
ChimeChimeSan Francisco, CA

Software Engineer, Machine Learning Platform

Build and operate Chime's ML platform on AWS, including distributed training systems, feature stores, data pipelines, and CI/CD tooling. Partner with ML teams to improve reliability, observability, and developer experience for production models.

187k – 259k/yr
Hybrid5+ YOEML Engineering

About the role

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 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

Requirements

  • 5+ years of experience in ML infrastructure, platform engineering, or production ML systems
  • Knowledge of the machine learning model development lifecycle (data preprocessing, model training, evaluation, 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
  • Experience with containerization technologies (Docker, Kubernetes)
  • Knowledge of cloud platforms (AWS) and distributed computing frameworks (Spark, Ray)
  • Experience with GPU programming (CUDA) and GPU optimization
  • Strong programming skills in Python, Go, Scala, Java or similar languages
  • Familiarity with infrastructure-as-code (Terraform, CloudFormation)
  • Solid understanding of software engineering fundamentals (testing, version control, code review, observability)

Nice-to-Haves

  • 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)
  • Experience supporting ML lifecycle workflows
  • Knowledge of ML experimentation platforms and model governance practices

Skills

PythonGoScalaJavaAWSKubernetesDockerTerraformRaySparkKafkaFlinkCUDA

Similar roles

ML Engineering jobs
EigenLayer

Agentic AI Engineer

EigenLayerSeattle, WA +1

Builds production-ready agentic AI systems including runtimes, orchestration, reliability, observability, and integrations with LLMs/APIs. Requires strong backend experience, shipped agent/LLM systems, and production reliability expertise.

187k – 253k/yr
RemoteML Engineering
Luma AI

Software Engineer, Inference

Luma AIPalo Alto, CA

Develops and optimizes inference engines for multimodal AI models, integrating new architectures, building scheduling systems, and managing large-scale GPU deployments. Requires strong Python, model serving frameworks like PyTorch/vLLM, and Kubernetes expertise.

188k – 395k/yr
HybridML Engineering
Luma AI

Software Engineer, ML Platform

Luma AIPalo Alto, CA

Builds foundational ML platform infrastructure including model serving pipelines, GPU scheduling systems, and CI/CD for large-scale multimodal AI models. Requires 5+ years in distributed systems with expertise in Python, Kubernetes, and AWS.

188k – 395k/yr
Hybrid5+ YOEML Engineering
Luma AI

Research Scientist / Engineer – Training Infrastructure

Luma AIPalo Alto, CA

Builds and optimizes distributed training infrastructure for large-scale multimodal AI models across thousands of GPUs. Requires deep expertise in PyTorch, CUDA, parallelization techniques, and GPU clusters.

188k – 395k/yr
HybridML Engineering
Reddit

Machine Learning Engineer, Ads Optimization & Ads Marketplace Quality

RedditUnited States

Build and evolve auction, bidding, and budgeting ML systems that power Reddit Ads. Design optimization algorithms balancing advertiser performance, user experience, and marketplace efficiency.

186k – 303k/yr
Remote3+ YOEML Engineering