Senior Research Engineer, LLM Training & Post-Training

Lightning AI
San Francisco, California, United States
Full Time
Hybrid
Senior
USD 165,000 – 310,000
Postgraduate Degree
Technology Science Rnd Non Medical

Posted 3 days ago

Apply to this job and hundreds more automatically

Get Started
Excellent 4.7

Job summary

Lightning AI seeks a Senior Research Engineer experienced in transformer-based language models, focusing on training, fine-tuning, evaluating, and deploying large language models within a fast-paced research and production environment. Responsibilities include optimizing training pipelines, improving model quality through various advanced techniques, building infrastructure, and collaborating across teams and with customers.

Responsibilities

Design, build, and optimize training and post-training pipelines for large language models while improving model quality and training efficiency. Collaborate with research, infrastructure, and platform engineering teams to translate cutting-edge research into production-ready AI systems.

Qualifications

distributed training fine-tuning model evaluation multi-GPU systems optimization techniques Python programming PyTorch research tooling training infrastructure transformer models

Requires significant experience in training, fine-tuning, and evaluating transformer-based language models using PyTorch and distributed systems. Candidates must possess strong software engineering fundamentals and the ability to solve complex performance bottlenecks in multi-GPU environments.

English

Benefits

  • Comprehensive Health Coverage
  • Meaningful Equity
  • Retirement Savings
  • Flexible Time Off
  • Company-Wide Winter Break
  • Paid Parental & Family Leave
  • Professional Development
  • Wellness Benefits
  • Sabbatical Program
  • Flexible Work
  • In-Office Meals

Company

Lightning AI

Lightning AI

Software Development
Location
San Francisco, California, United States

Apply to this job and hundreds more automatically

Get Started
Excellent 4.7
Pete DeQuarto avatar
Pete DeQuarto

AIApply is a total game-changer. Thanks to Auto-Apply, I have been applying to jobs I would never even have come across otherwise!

Akarshak Tanwar avatar
Akarshak Tanwar

In one week I applied to over 150 jobs and locked in 2 confirmed interviews, with a support team that was quick to respond. The auto-apply feature is fascinating and a real pleasure to use!

JC
Julian Carlin

It really worked. In under 48 hours it had applied to 100 jobs on my behalf without any time or effort from me. I am looking forward to seeing how it keeps improving.