Ai Engineer • Senior

Who you are

You are a highly skilled AI Optimization Engineer with a strong background in optimizing training and inference workloads on compute accelerators. You have experience working with the Lightning Thunder compiler and the broader PyTorch Lightning ecosystem, enabling you to take ideas from research to production efficiently. Your expertise in AI and machine learning allows you to innovate and create better products that serve a diverse range of clients, from solo researchers to large enterprises.

You thrive in a fast-paced environment and are committed to excellence in your craft. You understand the importance of collaboration and focus, ensuring that you complete one goal at a time with precision. Your ability to balance work and life contributes to sustained performance, and you take pride in mastering the details of your work. You are passionate about simplifying complex processes and driving innovation through minimalism.

What you'll do

In this role, you will be responsible for optimizing AI training and inference workloads, leveraging your expertise in compute accelerators and clusters. You will work closely with cross-functional teams to enhance the performance of the Lightning Thunder compiler, ensuring that it meets the needs of developers and researchers alike. Your contributions will directly impact the efficiency and effectiveness of AI systems deployed through the PyTorch Lightning platform.

You will collaborate with other engineers to identify bottlenecks in the current systems and propose innovative solutions that enhance performance. Your role will involve conducting experiments, analyzing results, and iterating on your findings to continuously improve the optimization processes. You will also have the opportunity to mentor junior engineers, sharing your knowledge and fostering a culture of learning within the team.

What we offer

At Lightning AI, we provide a dynamic work environment where you can thrive and contribute to cutting-edge AI solutions. We value speed, focus, balance, craftsmanship, and minimalism in our approach to innovation. You will have the chance to work with a talented team dedicated to pushing the boundaries of AI technology. We encourage you to apply even if your experience doesn't match every requirement, as we believe in the potential of diverse backgrounds and perspectives.