Full-Time

Machine Learning Engineer

Luma AI

Luma AI

11-50 employees

Enabling photorealistic 3D capture and mixed-reality experiences

Data & Analytics
AI & Machine Learning

Junior, Mid, Senior

Palo Alto, CA, USA

Required Skills
Kubernetes
Python
Pytorch
Docker
NumPy
Requirements
  • Strong programming skills in Python
  • Deep understanding of PyTorch, NumPy, and basic libraries for working with images and structured data
  • Experience with filtering and preparing training data
  • Experience with large model training
  • Experience working with large distributed systems like SLURM, Ray, or similar technologies
  • Experience with deploying models (building docker images, Kubernetes basics)
  • Knowledge of graphics fundamentals, 3D formats, tools (e.g. Blender/UE4), and/or 3D-related Python libraries
  • Ability to use and implement new models in codebases like diffusers, transformers
Responsibilities
  • Designing, implementing, and improving large-scale distributed machine learning systems
  • Writing bug-free machine learning code
  • Developing the underlying models
  • Rapidly building and adding features to CLI tools, dashboards for visualization, comparison, and metric tracking to accelerate research

Company Stage

Series B

Total Funding

$72.8M

Headquarters

, California

Founded

2021

Growth & Insights
Headcount

6 month growth

68%

1 year growth

200%

2 year growth

740%