Compiler Engineer
Palo Alto, California, United States
Full Time Senior-level / Expert USD 120K - 250K
Lightning AI
The all-in-one platform for AI development. Code together. Prototype. Train. Scale. Serve. From your browser - with zero setup. From the creators of PyTorch Lightning.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.
Focusing on simplicity, sustainability, modularity, and extensibility, Lightning AI streamlines the lifecycle of machine learning development to expand widespread AI adoption. Its aim is to enable individual and enterprise users to build deployment-ready AI tools without having to hire experts or sink resources into in-house infrastructure.
Lightning AI is a company with offices in New York City, Palo Alto, and London backed by $58.6 million in funding from Coatue, Index Ventures, Bain Capital Ventures, and Firstminute.
Our ValuesWe are building a company of people that care deeply about the impact of AI on our society. We value:
- Integrity: AI is bound to change how we interact with computers, with each other, and the world—for the better. We are committed to building an inclusive company and developing ethical AI that improves humanity.
- Perseverance: Be bold, take the initiative, and make decisions with confidence. We empower our community, clients, and colleagues to tackle the impossible.
- Curiosity: We are genuinely interested in finding solutions to advance the state of AI research and foster its application in novel products. It's not about whether we can solve the problem, but how.
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
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)
We champion equal pay for work of equal or comparable value based on the level of skill, effort, responsibility, and impact of any given role. We use Carta's Total Compensation tool to enable data-driven decision-making around benchmarks, levels, and planning. We are confident that our compensation practices are equitable, explainable, fair, and competitive, and in the spirit of transparency, we provide a compensation bands for each open role.
We offer competitive base salaries and stock options with a 25% one year cliff and monthly vesting thereafter. For our international employees, we work with Velocity Global to pay you in your local currency and provide equitable benefits across the globe.
In the US, we offer:
- 90% monthly premium contributions towards medical and 100% monthly premium contributions towards select dental and vision plans for you and your dependents
- Life and AD&D insurance
- Flexible paid time off plus 1 week of winter closure
- Generous paid family leave benefits
- $500 monthly meal reimbursement, including groceries & food delivery services
- $1,000 home office stipend
- $1,000 annual learning & development stipend
- 100% Citibike membership (NYC only)
- $45/month gym membership
- Additional various medical and mental health services
Tags: Architecture CUDA Deep Learning JAX Machine Learning Open Source PyTorch Research
Perks/benefits: Career development Competitive pay Equity / stock options Fitness / gym Flex vacation Health care Home office stipend Medical leave Transparency
More jobs like this
Explore more AI, ML, Data Science career opportunities
Find even more open roles in Artificial Intelligence (AI), Machine Learning (ML), Natural Language Processing (NLP), Computer Vision (CV), Data Engineering, Data Analytics, Big Data, and Data Science in general - ordered by popularity of job title or skills, toolset and products used - below.
- Open Research Scientist jobs
- Open Data Science Manager jobs
- Open Junior Data Analyst jobs
- Open Business Data Analyst jobs
- Open Data Scientist II jobs
- Open Principal Data Scientist jobs
- Open BI Analyst jobs
- Open Sr Data Engineer jobs
- Open Business Intelligence Engineer jobs
- Open Data Science Intern jobs
- Open Sr. Data Scientist jobs
- Open Senior Business Intelligence Analyst jobs
- Open Lead Data Analyst jobs
- Open Software Engineer, Machine Learning jobs
- Open Azure Data Engineer jobs
- Open Junior Data Scientist jobs
- Open MLOps Engineer jobs
- Open Manager, Data Engineering jobs
- Open Data Engineer III jobs
- Open Data Analytics Engineer jobs
- Open Marketing Data Analyst jobs
- Open Junior Data Engineer jobs
- Open Data Engineering Manager jobs
- Open Product Data Analyst jobs
- Open Data Analyst II jobs
- Open Privacy-related jobs
- Open GCP-related jobs
- Open Tableau-related jobs
- Open Excel-related jobs
- Open ML models-related jobs
- Open Data pipelines-related jobs
- Open APIs-related jobs
- Open PhD-related jobs
- Open PyTorch-related jobs
- Open Finance-related jobs
- Open LLMs-related jobs
- Open TensorFlow-related jobs
- Open Deep Learning-related jobs
- Open Data visualization-related jobs
- Open Consulting-related jobs
- Open Business Intelligence-related jobs
- Open Generative AI-related jobs
- Open CI/CD-related jobs
- Open Data governance-related jobs
- Open NLP-related jobs
- Open DevOps-related jobs
- Open Kubernetes-related jobs
- Open Docker-related jobs
- Open Git-related jobs
- Open Hadoop-related jobs