Research/Compiler Engineer

Tbwa Chiat/Day Inc
London
1 year ago
Applications closed

Related Jobs

View all jobs

Senior Machine Learning Applications and Compiler Engineer

Senior Machine Learning Applications and Compiler Engineer

CargoWise Master Data Analyst

Data Science Specialist

Research Assistant/Associate in Data Science and Computational Neuroscience

Research Fellow - Data Engineering & ML for Battery

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.

We are on a mission to simplify AI development, making it accessible to everyone—from solo researchers to large enterprises. By removing the complexity of building and deploying AI tools, we empower innovators to focus on solving real-world problems. Our platform is built to scale with the latest AI advancements while staying intuitive and adaptable, so you can bring your ideas to life.

We have offices in New York City, Palo Alto, and London and are backed by investors such as Coatue, Index Ventures, Bain Capital Ventures, and Firstminute.

Our Values

Move Fast: We act with speed and precision, breaking down big challenges into achievable steps.

Focus: We complete one goal at a time with care, collaborating as a team to deliver features with precision.

Balance: Sustained performance comes from rest and recovery. We ensure a healthy work-life balance to keep you at your best.

Craftsmanship: Innovation through excellence. Every detail matters, and we take pride in mastering our craft.

Minimal: Simplicity drives our innovation. We eliminate complexity through discipline and focus on what truly matters.

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 one of our strategic partners, 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.

What you’ll need

  • Strong experience with deep learning frameworks like PyTorch, JAX, or TensorFlow.
  • Expertise in compiler development or optimizations in distributed training and inference workflows is highly valued.
  • Proven track record contributing to open-source projects, especially in machine learning or high-performance computing. Experience collaborating with external partners is a plus.
  • Hands-on experience in model optimization, with a focus on maximizing performance, efficiency, and scalability in large-scale or distributed training setups.
  • Passion for engaging with open-source communities, including experience supporting users and advocating for project adoption.
  • Strong communication and collaboration skills for working within a close-knit, high-impact team environment.
  • Bachelor's degree in Computer Science, Engineering, or related field. Preferred Master's or PhD in machine learning and related areas.

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:

  • Medical, dental and vision.
  • 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).
  • Additional various medical and mental health services.

At Lightning AI, we are committed to fostering an inclusive and diverse workplace. We believe that diverse teams drive innovation and create better products. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity, national origin, age, disability, veteran status, or any other protected characteristic. We are dedicated to building a culture where everyone can thrive and contribute to their fullest potential.

Apply for this job

* indicates a required field

First Name *

Last Name *

Email *

Phone *

Resume/CV *

Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

Can you work from our NYC, London, or Palo Alto offices 2-3 days per week? * Select...

Have you worked for a Startup in your career? If so, how many years of Startup experience do you have? *

#J-18808-Ljbffr

Subscribe to Future Tech Insights for the latest jobs & insights, direct to your inbox.

By subscribing, you agree to our privacy policy and terms of service.

Industry Insights

Discover insightful articles, industry insights, expert tips, and curated resources.

Careers

How Many Machine Learning Tools Do You Need to Know to Get a Machine Learning Job?

Machine learning is one of the most exciting and rapidly growing areas of tech. But for job seekers it can also feel like a maze of tools, frameworks and platforms. One job advert wants TensorFlow and Keras. Another mentions PyTorch, scikit-learn and Spark. A third lists Mlflow, Docker, Kubernetes and more. With so many names out there, it’s easy to fall into the trap of thinking you must learn everything just to be competitive. Here’s the honest truth most machine learning hiring managers won’t say out loud: 👉 They don’t hire you because you know every tool. They hire you because you can solve real problems with the tools you know. Tools are important — no doubt — but context, judgement and outcomes matter far more. So how many machine learning tools do you actually need to know to get a job? For most job seekers, the real number is far smaller than you think — and more logically grouped. This guide breaks down exactly what employers expect, which tools are core, which are role-specific, and how to structure your learning for real career results.

Jobs

What Hiring Managers Look for First in Machine Learning Job Applications (UK Guide)

Whether you’re applying for machine learning engineer, applied scientist, research scientist, ML Ops or data scientist roles, hiring managers scan applications quickly — often making decisions before they’ve read beyond the top third of your CV. In the competitive UK market, it’s not enough to list skills. You must send clear signals of relevance, delivery, impact, reasoning and readiness for production — and do it within the first few lines of your CV or portfolio. This guide walks you through exactly what hiring managers look for first in machine learning applications, how they evaluate CVs and portfolios, and what you can do to improve your chances of getting shortlisted at every stage — from your CV and LinkedIn profile to your cover letter and project portfolio.

Careers Jobs

MLOps Jobs in the UK: The Complete Career Guide for Machine Learning Professionals

Machine learning has moved from experimentation to production at scale. As a result, MLOps jobs have become some of the most in-demand and best-paid roles in the UK tech market. For job seekers with experience in machine learning, data science, software engineering or cloud infrastructure, MLOps represents a powerful career pivot or progression. This guide is designed to help you understand what MLOps roles involve, which skills employers are hiring for, how to transition into MLOps, salary expectations in the UK, and how to land your next role using specialist platforms like MachineLearningJobs.co.uk.

🍪 We use cookies to enhance your browsing experience on our website. By continuing to use this site, you consent to the use of cookies as described in our Cookie Policy. You can review our Cookie Policy for more information.