Research/Compiler Engineer

1 week ago


Palo Alto, United States Lightning AI Full time
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


  • Palo Alto, California, United States Lightning AI Full time

    About UsLightning AI is a pioneering company redefining the development of artificial intelligence products. Our mission is to simplify the lifecycle of machine learning development, making it accessible to individuals and enterprises alike.Our ValuesWe are committed to building an inclusive company and developing ethical AI that improves humanity. Our...


  • Palo Alto, California, United States RI Research Instruments GmbH Full time

    RI Research Instruments GmbH is dedicated to pioneering multimodal AI technologies that enhance human creativity and capabilities. We recognize that true intelligence requires a multimodal approach. Our focus is on advancing beyond traditional language models to develop systems that can perceive, comprehend, and interact with the world around us. We are in...


  • Palo Alto, California, United States RI Research Instruments GmbH Full time

    RI Research Instruments GmbH is dedicated to advancing multimodal artificial intelligence to enhance human creativity and capabilities. We recognize that multimodality is essential for true intelligence. Our goal is to transcend traditional language models by integrating vision into our systems. We are focused on developing and scaling multimodal foundation...


  • Palo Alto, California, United States Electric Power Research Institute Full time

    Job SummaryThis is a challenging role that requires a highly skilled and experienced professional to conduct research in complex technical or scientific fields. The successful candidate will be responsible for leading technical activities, providing guidance to junior staff, and communicating complex strategies and results to subject matter experts and...

  • Research Engineer

    3 months ago


    Palo Alto, United States Acceler8 Talent Full time

    Join US as a Founding Machine Learning Research EngineerAre you passionate about advancing AI systems and tackling complex challenges in machine learning? We are seeking enthusiastic individuals to join our pioneering team as Founding ML Research Engineers. This role offers both junior and senior opportunities, allowing individuals at different stages of...

  • Research Engineer

    3 months ago


    Palo Alto, United States Acceler8 Talent Full time

    Join US as a Founding Machine Learning Research EngineerAre you passionate about advancing AI systems and tackling complex challenges in machine learning? We are seeking enthusiastic individuals to join our pioneering team as Founding ML Research Engineers. This role offers both junior and senior opportunities, allowing individuals at different stages of...


  • Palo Alto, California, United States Centrillion Technology Holdings Full time

    Centrillion Technology Holdings is at the forefront of innovative genomic solutions, harnessing engineering and biochemical advancements to enhance discovery and unveil the complexities of the genome that have remained hidden due to existing technological constraints. Our dedicated team of scientists and engineers is focused on making a new era of genomic...


  • Palo Alto, California, United States Electric Power Research Institute Full time

    Job Title:DERMS & Microgrid Staff/Researcher IVLocation:EPRI Knoxville Office, Palo Alto, CA, Remote/Home BasedJob Summary and Description:EPRI's DERMS team is seeking to fill a Sr Research Scientist position to lead research on emerging technologies in the smart grid including smart inverters, DER Management Systems and Microgrid Controls. The candidate...


  • Palo Alto, California, United States PsiQuantum Full time

    Position: Photonics Research Engineer at PsiQuantumAt PsiQuantum, we are at the forefront of quantum computing innovation, committed to transforming industries globally. We are currently in search of a skilled Photonics Research Engineer to become an integral part of our innovative team.Role Overview:Develop, execute, and refine advanced experimental...


  • Palo Alto, United States Electric Power Research Institute Full time

    Job Title: Nuclear Fuel Materials Expert - Senior Researcher Location: Charlotte, NC, Palo Alto, CA, Remote/Home Based Job Summary and Description: Conducts research in complex, high impact technical or scientific fields. Key Responsibilities and Requirements: -Works independently on technical activities without direct...


  • Palo Alto, United States Electric Power Research Institute Full time

    Job Title: DERMS & Microgrid Staff/Researcher IV Location: EPRI Knoxville Office, Palo Alto, CA, Remote/Home Based Job Summary and Description: EPRI’s DERMS team is seeking to fill a Sr Research Scientist position to lead research on emerging technologies in the smart grid including smart inverters, DER Management Systems and Microgrid...


  • Palo Alto, United States Electric Power Research Institute Full time

    Description : Conducts research in complex, high impact technical or scientific fields. Key Responsibilities and Requirements: -Works independently on technical activities without direct supervision. -Demonstrates responsibility for highly specialized technical objectives or problems where the relationships between cause and effect are difficult to...


  • Palo Alto, United States Tesla Full time

    What to ExpectAs a member of the Dojo correctness and reliability team, you will be responsible for building program analysis tools (fuzzers, compiler instrumentation passes, run-time monitoring components) for our in-house supercomputer. Join a small team of experienced developers in automating software and hardware testing.  What You’ll DoTake...


  • Palo Alto, California, United States PsiQuantum Full time

    About PsiQuantum:At PsiQuantum, we are on a transformative journey to revolutionize the computing landscape through quantum technology. Our mission is to develop the world's first practical quantum computer, which will significantly enhance computational capabilities across various industries.Position Overview:We are seeking a talented Photonics Research...


  • Palo Alto, California, United States Tesla, Inc. Full time

    Foundation Models Research EngineerTesla is in search of outstanding software engineers to advance the development of AI's foundation models. You will collaborate with a select group of elite deep learning professionals to create cutting-edge neural networks and explore the frontiers of AI research and innovation. Your contributions will facilitate the...


  • Palo Alto, California, United States Tesla, Inc. Full time

    Foundation Models Research EngineerTesla is on the lookout for outstanding software engineers to contribute to the development of AI's foundation models. You will collaborate with a select group of elite deep learning specialists to create cutting-edge neural networks and explore the frontiers of AI research and innovation. Your contributions will facilitate...


  • Palo Alto, California, United States Tesla, Inc. Full time

    Foundation Models Research EngineerTesla is on the lookout for outstanding software engineers to contribute to the development of AI's foundation models. You will collaborate with a select group of elite deep learning professionals to create cutting-edge neural networks and expand the horizons of AI research and innovation. Your contributions will facilitate...


  • Palo Alto, California, United States Tesla, Inc. Full time

    Research Engineer, Foundation Models, Self-DrivingTesla is on the lookout for outstanding software engineers to develop the foundation models for Tesla AI. You will collaborate with a select group of elite deep learning specialists to create cutting-edge neural networks and explore the frontiers of AI research and innovation. Your contributions will...


  • Palo Alto, California, United States Electric Power Research Institute Full time

    Position SummaryThe Electric Power Research Institute (EPRI) is seeking a **Principal Software Engineer** within the **Transmission Operations and Planning** division. This position is pivotal in the creation and execution of innovative software solutions tailored for power systems. You will lead a skilled team, oversee software policy alignment, and engage...


  • Palo Alto, United States Tesla Full time

    What to ExpectAs a ML Software Engineer within Dojo, you will play a crucial role in bridging the gap between our cutting-edge Dojo training accelerator and the neural networks developed by our Autopilot ML team. Collaborate closely with world-class ML Researchers, Compiler and Hardware Engineers to tackle unique challenges at the intersection of AI and ML...