Research Engineer

United Kingdom·San Francisco,London,Seattle+1 moremid

OtherResearch EngineerAi Research EngineerData & AI

5 views0 saves0 applied

Apply Now

Quick Summary

Key Responsibilities

Develop performance-oriented model optimizations at multiple levels: Graph-level (e.g., operator fusion, kernel scheduling, memory planning) Kernel-level (CUDA, Triton, custom operators for specialized hardware) System-level (distributed training…

Requirements Summary

We are seeking a highly skilled Research Engineer to work on optimizing training and inference workloads on compute accelerators and clusters, through the Lightning Thunder compiler and the broader PyTorch Lightning ecosystem.

Technical Tools

pytorchab-testingapi-designci-cddeep-learningdistributed-systemsmachine-learningperformance-optimization

Lightning AI is the company behind PyTorch Lightning. Founded in 2019, we build an end-to-end platform for developing, training, and deploying AI systems—designed to take ideas from research to production with less friction.

Through our merger with Voltage Park, a neocloud and AI Factory, Lightning AI combines developer-first software with cost-efficient, large-scale compute. Teams get the tools they need for experimentation, training, and production inference, with security, observability, and control built in.

We serve solo researchers, startups, and large enterprises. Lightning AI operates globally with offices in New York City, San Francisco, Seattle, and London, and is backed by Coatue, Index Ventures, Bain Capital Ventures, and Firstminute.

Responsibilities

~1 min read

→Comprehensive medical, dental and vision coverage (U.S.); Private medical and dental insurance (U.K.)
→Retirement and financial wellness support (U.S.); Pension contribution (U.K.)
→Generous paid time off, plus holidays
→Paid parental leave
→Professional development support
→Wellness and work-from-home stipends
→Flexible work environment

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.