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
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
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
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
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.
Build and evolve auction, bidding, and budgeting ML systems that power Reddit Ads. Design optimization algorithms balancing advertiser performance, user experience, and marketplace efficiency.