lightningai
Research Engineer
At a Glance
- Location
- New York, New York, United States; San Francisco, California, United States
- Compensation
- salary range for this role is $120,000-$250,000. What you’ll do Develop perfo
- Posted
- 2026-02-25T12:08:13-05:00
Key Requirements
Required Skills
Domain Knowledge
- Engineering
- Insurance
- Medical
Requirements
Lightning Thunder compiler
PyTorch Lightning ecosystem
This role sits at the intersection of
deep learning research, compiler development, and large-scale system optimization
You’ll be shaping technology that pushes the boundaries of model performance and efficiency, creating foundational software that will impact the entire machine learning ecosystem.
You will be joining the Engineering Team and report to our Tech Lead.
Responsibilities
Develop performance-oriented model optimizations
Kernel-level (CUDA, Triton, custom operators for specialized hardware)
System-level (distributed training across GPUs/TPUs, inference serving at scale)
Advance the Thunder compiler
by building optimization passes, graph transformations, and integration hooks to accelerate training and inference workloads.
Work across the software stack