Job Description

Lightning AI is seeking a skilled AI Optimization Leader to optimize training and inference workloads using the Lightning stack. This role involves deep learning research, compiler development, and large-scale system optimization, impacting customer success and revenue generation. The position requires a hybrid work model with in-office requirements.

Responsibilities

• Own technical direction of model optimization efforts
• Advance compiler technology
• Ensure optimizations are accessible to end users
• Design and implement profiling and debugging tools
• Collaborate with hardware vendors
• Contribute to open-source projects
• Engage with researchers and engineers
• Work cross-functionally with product and engineering teams

Requirements

• Strong expertise with deep learning frameworks
• Hands-on experience profiling models
• Experience with model optimization techniques
• Deep understanding of compiler internals
• Experience with CUDA, Triton, or GPU programming
• Knowledge of distributed computing
• Familiarity with software engineering practices
• Proven track record contributing to open-source projects
• Excellent collaboration and communication skills
• Bachelor’s degree in Computer Science or related field

Benefits

• No benefits