Research Engineer, ML Compiler & Systems Optimization

Lightning AI
San Francisco, US
Hybrid

Job Description

Lightning AI is seeking a Research Engineer to optimize training and inference workloads using the Lightning Thunder compiler. This hybrid role requires expertise in deep learning with PyTorch and model optimization techniques. Responsibilities span from developing model optimizations to collaborating with hardware vendors. Candidates should possess strong communication skills and preferably a Master’s or PhD in relevant fields. Base salary is $180,000–$250,000, along with competitive benefits and a supportive work environment. #J-18808-Ljbffr

Skills & Requirements

Technical Skills

Deep learningPytorchModel optimization techniquesCommunicationMachine learningEngineering

Salary

$180,000 - $250,000

year

Employment Type

FULL TIME

Level

mid

Posted

4/29/2026

Apply Now

You will be redirected to Lightning AI's application portal.

Sign in and we'll score your resume against this role.