ML Engineer - Inference Serving

Posted 13 Days Ago
2 Locations
Hybrid
188K-395K Annually
Mid level
Artificial Intelligence • Software
Luma AI’s mission is to build Multimodal AGI
The Role
The ML Engineer will integrate model architectures, optimize deployment workflows, maintain CI/CD pipelines, and ensure reliability of inference services across large-scale systems.
Summary Generated by Built In
About Luma AI
Luma's mission is to build multimodal AI to expand human imagination and capabilities. We believe that multimodality is critical for intelligence. To go beyond language models and build more aware, capable and useful systems, the next step function change will come from vision. So we are working on training and scaling up multimodal foundation models for systems that can see and understand, show and explain, and eventually interact with our world to effect change.

Luma’s mission is to build multimodal AI to expand human imagination and capabilities.
We believe that multimodality is critical for intelligence. To go beyond language models and build more aware, capable and useful systems, the next step function change will come from vision. We are working on training and scaling up multimodal foundation models for systems that can see and understand, show and explain, and eventually interact with our world to affect change. We know we are not going to reach our goal with reliable & scalable infrastructure, which is going to become the differentiating factor between success and failure.

Role & Responsibilities
  • Ship new model architectures by integrating them into our inference engine
  • Collaborate closely across research, engineering and infrastructure to streamline and optimize model efficiency and deployments
  • Build internal tooling to measure, profile, and track the lifetime of inference jobs and workflows
  • Automate, test and maintain our inference services to ensure maximum uptime and reliability
  • Optimize deployment workflows to scale across thousands of machines
  • Manage and optimize our inference workloads across different clusters & hardware providers
  • Build sophisticated scheduling systems to optimally leverage our expensive GPU resources while meeting internal SLOs
  • Build and maintain CI/CD pipelines for processing/optimizing model checkpoints, platform components, and SDKs for internal teams to integrate into our products/internal tooling

Background
  • Strong Python and system architecture skills
  • Experience with model deployment using PyTorch, Huggingface, vLLM, SGLang, tensorRT-LLM, or similar
  • Experience with queues, scheduling, traffic-control, fleet management at scale
  • Experience with Linux, Docker, and Kubernetes
  • Bonus points: 
    • Experience with modern networking stacks, including RDMA (RoCE, Infiniband, NVLink)
    • Experience with high performance large scale ML systems (>100 GPUs)
    • Experience with FFmpeg and multimedia processing

Example Projects
  • Create a resilient artifact store that manages all checkpoints across multiple versions of multiple models
  • Enable hotswapping of models for our GPU workers based on live traffic patterns
  • Build a robust queueing system for our jobs that take into account cluster availability and user priority
  • Architect a e2e model serving deployment pipeline for a custom vendor
  • Integrate our inference stack into an online reinforcement learning pipeline
  • Regression & precision testing across different hardware platforms
  • Building a full tracing system to trace the end-to-end lifetime of any inference workload

Tech stackMust have
  • Python
  • Redis
  • S3-compatible Storage
  • Model serving (one of: PyTorch, vLLM, SGLang, Huggingface)
  • Understanding of large-scale orchestration, deployment, scheduling (via Kubernetes or similar)
Nice to have
  • CUDA
  • FFmpeg

Compensation
The base pay range for this role is $187,500 – $395,000 per year.

Top Skills

Cuda
Ffmpeg
Huggingface
Kubernetes
Python
PyTorch
Redis
S3-Compatible Storage
Sglang
Vllm
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco Bay Area, CA
246 Employees
Year Founded: 2021

What We Do

Luma AI’s mission is to build Multimodal AGI: AI that can generate, understand, and operate in the physical world. We develop multimodal models across video, 3D, and generative media, and ship them in products like Dream Machine to help creators and teams turn ideas into compelling visuals—fast.

Similar Jobs

Anduril Logo Anduril

Technical Operations Engineer, Bolt

Aerospace • Artificial Intelligence • Hardware • Robotics • Security • Software • Defense
In-Office
Costa Mesa, CA, USA
6000 Employees
113K-1M Annually

CrowdStrike Logo CrowdStrike

Infrastructure Engineer

Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Remote or Hybrid
2 Locations
10000 Employees

CrowdStrike Logo CrowdStrike

Patent Attorney (Remote)

Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Remote or Hybrid
USA
10000 Employees

CrowdStrike Logo CrowdStrike

Security Engineer

Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Remote or Hybrid
2 Locations
10000 Employees

Similar Companies Hiring

Fairly Even Thumbnail
Software • Sales • Robotics • Other • Hospitality • Hardware
New York, NY
Bellagent Thumbnail
Artificial Intelligence • Machine Learning • Business Intelligence • Generative AI
Chicago, IL
20 Employees
Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account