Lightning AI is the company reimagining the way AI is built. After creating and releasing PyTorch Lightning in 2019, Lightning AI was launched to reshape the development of artificial intelligence products for commercial and academic use. We are on a mission to simplify AI development, making it accessible to everyone—from solo researchers to large enterprises. By removing the complexity of building and deploying AI tools, we empower innovators to focus on solving real-world problems. Our platform is built to scale with the latest AI advancements while staying intuitive and adaptable, so you can bring your ideas to life. We have offices in New York City, San Francisco, and London and are backed by investors such as Coatue, Index Ventures, Bain Capital Ventures, and Firstminute. We are seeking a highly skilled AI Optimization Leader to work on optimizing training and inference workloads on compute accelerators and clusters, through the Lightning stack, including the Thunder compiler and the broader PyTorch Lightning ecosystem. This role sits at the intersection of deep learning research, compiler development, and large-scale system optimization. This role is also a core pillar of Lightning’s business strategy. You will be at the forefront of how we service high impact customer deals that rely on advanced optimization services, acting as a key technical and strategic leader for one of the company's major revenue-driving functions. Your work will directly influence customer success, deal structure, and our ability to scale optimization-centric offerings as a competitive advantage. You will be joining the Engineering Team and report to our Director of Engineering. This is a hybrid role based in our San Francisco office with in-office requirements of 2 days per week. The salary range for this role is $225,000-$275,000.