companyLightning AI logo

Research Engineer

Lightning AILondon, England, United Kingdom; New York, New York, United States; San Francisco, California, United States
Hybrid Full-time $180K/yr - $250K/yr

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Entry Level

Qualifications

The ideal candidate will possess a strong background in machine learning and systems optimization, with experience in compiler development. Familiarity with deep learning frameworks, particularly PyTorch, is essential. Candidates should demonstrate the ability to work collaboratively in a team-oriented environment and have a passion for driving innovation in AI technology.

About the job

About Lightning AI

Founded in 2019, Lightning AI is the driving force behind PyTorch Lightning. We create a comprehensive platform for the development, training, and deployment of AI systems, streamlining the journey from innovative research to impactful production.

Our merger with Voltage Park, a neocloud and AI factory, enhances our offerings by integrating developer-centric software with efficient, large-scale computing resources. We provide teams with essential tools for experimentation, training, and production inference, ensuring security, observability, and control are integral to our solutions.

We cater to individual researchers, startups, and established enterprises, with a global presence including offices in New York City, San Francisco, Seattle, and London. Lightning AI is supported by esteemed investors such as Coatue, Index Ventures, Bain Capital Ventures, and Firstminute.

Our Values

  • Move Fast: We prioritize speed and precision, deconstructing complex challenges into manageable tasks.

  • Focus: We tackle one objective at a time with care, working collaboratively to deliver features accurately.

  • Balance: We believe in maintaining a healthy work-life balance, promoting sustained performance through rest and recovery.

  • Craftsmanship: We strive for excellence and take pride in the details that contribute to our innovation.

  • Minimal: We embrace simplicity in our innovations, focusing on what truly matters.

Role Overview

We are looking for an accomplished Research Engineer to enhance the optimization of training and inference workloads using compute accelerators and clusters, particularly through the Lightning Thunder compiler and the broader PyTorch Lightning ecosystem. This role exists at the intersection of deep learning research, compiler development, and large-scale system optimization. You will be instrumental in advancing technology that enhances model performance and efficiency, establishing essential software that will influence the entire machine learning landscape.

As part of the Engineering Team, you will report to our Tech Lead. This hybrid role is based in our offices located in New York City, San Francisco, or London.

About Lightning AI

Lightning AI, founded in 2019, is at the forefront of AI innovation, providing a seamless platform for developing and deploying AI systems. Our commitment to excellence is reflected in our partnerships with leading investors and our global presence.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.