Lead Research Engineer: AI Compiler & Scalable Training

Lightning AI
Seattle; Washington, US
Hybrid

Job Description

Lightning AI is seeking a Lead Research Engineer to optimize training and inference workloads using the Lightning Thunder compiler within the PyTorch Lightning ecosystem. This hybrid role is in San Francisco, New York City, or London, requiring 2 days in-office weekly. The ideal candidate has strong expertise in deep learning frameworks and model optimization techniques. Based on experience, salaries range from $225,000 to $275,000, along with comprehensive benefits including health insurance and paid family leave. #J-18808-Ljbffr

Skills & Requirements

Technical Skills

Deep learning frameworksModel optimization techniquesPytorch lightning ecosystemAiCompilerScalable training

Salary

$225,000 - $275,000

year

Employment Type

FULL TIME

Level

lead

Posted

4/30/2026

Apply Now

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

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