Compiler Engineer

Lightning AI is seeking a Compiler Engineer to work on the Thunder compiler and PyTorch Lightning stack. This role offers an opportunity to create groundbreaking technology that will transform the machine learning ecosystem. The ideal candidate will have deep experience in PyTorch, JAX, or other deep learning frameworks.

Key responsibilities include:

• Developing the Thunder compiler, an open-source project in collaboration with NVIDIA
• Engaging in performance-oriented model optimizations for distributed training and inference
• Developing optimized kernels in CUDA or Triton for specific use-cases
• Integrating Thunder throughout the PyTorch Lightning ecosystem
• Engaging with the community and supporting Thunder's adoption across the industry

Lightning AI values integrity, perseverance, and curiosity. They are committed to building an inclusive company and developing ethical AI that improves humanity. The company offers competitive compensation, stock options, and a comprehensive benefits package.

This role provides an exciting opportunity to work at the forefront of AI development, collaborating with industry leaders and contributing to open-source projects that will shape the future of machine learning.