AI Performance Optimization Engineer
Lightning AINew York, New York, United States • San Francisco, California, United States
Posted September 24, 2025
120000 - 240000 USD year
Full Time
About the Company
Lightning AI is an innovative platform for accelerating AI development from idea to deployment. It offers a user-friendly interface with zero setup, enabling seamless coding, prototyping, training on GPUs, and scaling infrastructure. Key features include persistent storage, multi-node support, and integration with popular AI frameworks like PyTorch Lightning, making it a distinctive solution for artificial intelligence, machine learning, deep learning, data science, and open-source projects.
Job Description
Lightning AI is seeking a highly skilled AI Optimization Engineer to optimize training and inference workloads on compute accelerators and clusters using the Lightning Thunder compiler and PyTorch Lightning. This role bridges deep learning research, compiler development, and system optimization, focusing on creating foundational software for the ML ecosystem. The position is hybrid, residing in either New York City or San Francisco, with in-office requirements of 2 days per week.
Requirements
- Develop performance-oriented model optimizations
- Work across the software stack
- Collaborate with hardware vendors
- Contribute to open-source projects
- Engage with researchers and engineers in the community
Benefits
- Competitive base salary
- Stock options
- Medical, dental and vision insurance
- Life and AD&D insurance
- Flexible paid time off
- Generous paid family leave
- Meal reimbursement
- Home office stipend
- Learning & development stipend
- Fitness perks
- Mental health services
