What we're looking for
We are looking for a software engineer to work directly on the Lightning Thunder compiler and the rest of the PyTorch Lightning stack. This is an opportunity to create groundbreaking technology that will transform the machine learning ecosystem.
What youβll do
- Develop the Thunder compiler, an open-source project developed in collaboration with NVIDIA, using your deep experience in PyTorch, JAX, or other deep learning frameworks
- Engage in performance-oriented model optimizations, around distributed training as well as inference
- Develop optimized kernels in CUDA or Triton to target specific use-cases
- Integrate Thunder throughout the PyTorch Lightning ecosystem
- Engage with the community and champion its growth
- Support the adoption of Thunder across the industry
- Work closely within the Lightning team as a strategic partner
How we evaluate your candidacy
Our interviews will not only give us insight into your skillset but give you insight into our company.
- Introduction: You will speak with our Recruiter about your background and what you are looking for in your next role. They will share some information with you about our team and the projects you will be involved in. If both parties decide that there is a match, we will proceed with a technical interview.
- Coding: Our second interview will be a coding interview with our CTO.
- Virtual Onsite: These interviews will cover your domain expertise, systems architecture and Culture.
- Offer: We determine your level based on your interview performance and make an offer based on geo-located salary bands updated on a monthly basis using Carta's Total Compensation tool.
- Compensation: $120,000 - $250,000 (depending on leveling)
Β
![Lightning AI](https://s3-recruiting.cdn.greenhouse.io/external_greenhouse_job_boards/logos/400/388/800/resized/ihatethis.jpg?1717708333)
Lightning AI
A fast-growing startup focused on building innovative solutions for data management and analytics
Other jobs at Lightning AI
Β
Β
Β
Β
Β
Β
Β
Β
Notifications about similar jobs
Get notifications to your inbox about new jobs that are similar to this one.
No spam. No ads. Unsubscribe anytime.
Similar jobs
Β
Β
Β
Β
Β
Β
Β
Β