SWE - Data Infrastructure jobs in United States
info-icon
This job has closed.
company-logo

Luma AI · 4 months ago

SWE - Data Infrastructure

positionPalo Alto, CA
timeFull-time
remoteOnsite
seniorityMid, Senior Level
money$220K/yr - $280K/yr

Luma AI is dedicated to building multimodal AI to enhance human imagination and capabilities. As a Data Infrastructure Engineer, you will be responsible for developing and scaling data infrastructure that supports high-throughput machine learning workflows, collaborating with ML researchers and product teams to create efficient data systems.

Artificial Intelligence (AI)Foundational AIGenerative AIVideoVideo Editing
check
H1B Sponsor Likelynote

Responsibilities

Build and maintain scalable data infrastructure for high-throughput machine learning workflows
Collaborate with ML researchers and product teams to ensure data systems meet evolving needs
Develop and optimize large-scale data pipelines and batch processing jobs
Contribute to the architecture and implementation of reliable, high-performance data platforms
Integrate open-source tools and continuously improve data infrastructure through monitoring and tuning
Participate in cross-functional projects to improve data reliability, scalability, and operational excellence
Support the evaluation and adoption of new programming languages and frameworks relevant to data infrastructure
Engage in continuous improvement of data infrastructure through monitoring, troubleshooting, and performance tuning
Collaborate with research & engineering teams to help define and refine best practices for data infrastructure development

Qualification

PythonDistributed computing frameworksData pipeline designData engineering at scaleLarge datasetsOpen source projectsProblem-solving skillsCollaborative mindset

Required

Proficiency in Python (or similar languages with willingness to learn Python) and experience with large-scale, high-throughput data infrastructure
Familiarity with distributed computing frameworks (e.g., Ray, Spark, Beam)
Ability to design and optimize data pipelines for ML research and internal teams
Strong problem-solving skills and understanding of data engineering at scale
Collaborative, product-focused mindset; comfortable in fast-paced environments
Experience sourcing, integrating, and optimizing data from diverse and large datasets
Comfortable working in a fast-paced, product-focused environment with a strong execution mindset
Open to candidates across seniority levels, from mid-level individual contributors to senior engineers and managers

Preferred

Prior experience working with complex data infrastructure or AI/ML platforms highly desirable
Experience with open source data infrastructure projects is a plus

Company

Luma AI

twittertwittertwitter
company-logo
Luma AI develops tools that let users generate photorealistic images and videos from text, image, or video prompts.
dateFounded in 2021
location
Palo Alto, California, USA
people-size11-50 employees
web-link
https://lumalabs.ai

H1B Sponsorship

Luma AI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (10)
2024 (3)

Funding

Current Stage
Growth Stage
Total Funding
$1.06B
Key Investors
HUMAINAndreessen HorowitzAmplify Partners
2025-11-19Series C· $900M
2024-12-06Series B· $90M
2024-01-09Series B· $43M

Leadership Team

leader-logo
Amit Jain
Co-Founder
linkedin

Recent News

MarketWatch
As AMD chases Nvidia, here’s how it’s positioning itself for the future of AI
2026-01-09
Morningstar.com
As AMD chases Nvidia, here's how it's positioning itself for the future of AI
2026-01-06
Business Wire
Luma AI Announces Ray3 Modify, a New Model for Hybrid-AI Workflows for Acting & Performances, Now Available in Dream Machine
2025-12-18
Company data provided by crunchbase