Research/Compiler Engineer

Lightning AI • New York City, NY

Company

Lightning AI

Location

New York City, NY

Type

Full Time

Job Description

Who We Are

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. Our 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 has offices in New York City, Palo Alto, and London, and is backed by $58.6 million in funding from Coatue, Index Ventures, Bain Capital Ventures, and Firstminute.

Our Values

We are building a company of people who care deeply about the impact of AI on our society. We value:

  • Integrity: AI is bound to change how we interact with computers, 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 also recognize the genuine lack of diversity in our industry, which has to change. One of our goals is to help drive that change. Lightning AI is fully committed to diversity, equity, and inclusion in all of its practices, especially when it comes to growing our team. Our culture promotes inclusion, and we embrace how rewarding it is to work with people from all walks of life.

What we're looking for

We are looking for a research 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
Benefits and Perks

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
Apply Now

Date Posted

09/16/2024

Views

0

Back to Job Listings ❤️Add To Job List Company Info View Company Reviews
Positive
Subjectivity Score: 0.8

Similar Jobs

AWS Alliance Driver, Director - PwC

Views in the last 30 days - 0

The text describes a role for an AWS Alliance Director at PwC The individual will lead the AWS Alliance across various sectors focusing on the overall...

View Details

Senior Software Engineer, Devices Automation - Block

Views in the last 30 days - 0

Square a company that has evolved since its inception in 2009 is seeking a Software Engineer with extensive experience in embedded devices and test en...

View Details

Principal Product Marketing Manager - Rokt

Views in the last 30 days - 0

mParticle by Rokt a leading customer data platform is seeking a Principal Product Marketing Manager The role involves driving market leadership creati...

View Details

Process Delivery Specialist - Lead to Cash - IBM

Views in the last 30 days - 0

The job posting is for an Order to Cash Analyst role in IBM Consulting The role involves ensuring accurate and timely application of customer payments...

View Details

Senior Data Engineer - Sortly

Views in the last 30 days - 0

Sortly is a successful distributed and remotefirst company offering a multidevice inventory management solution They are seeking a Data Engineer with ...

View Details

Content Strategy Manager (Hybrid) - Terra (terrahq.com)

Views in the last 30 days - 0

Terra is hiring a Content Strategy Manager with 7 years of experience in clientfacing agency work particularly in the health and finance sectors The r...

View Details