AI Performance Optimization Engineer

Lightning AI
San Francisco, NY
Category Research
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
]]>