Senior Research Engineer, LLM Training & Post-Training
Benefits
Unlimited PTO
Similar Jobs
See allStaff / Senior IC Post-Training Engineer
Black Forest Labs
Global
PyTorch
Reinforcement Learning
Senior Machine Learning Engineer, ML Efficiency
US
PyTorch
Machine Learning
GPU
Research Intern
Cohere
Global
Python
Machine Learning
NLP
Senior Machine Learning Infrastructure Engineer, Embedding Platform
US
Python
PyTorch
Deep Learning
Machine Learning Engineer - Large Language Models
Unknown
US
Large Language Models
PyTorch
VLLM
Who We Are:
- Lightning AI is the company behind PyTorch Lightning, building an end-to-end platform for AI development and deployment.
- Founded in 2019, we operate globally with offices in New York, San Francisco, Seattle, and London, backed by Coatue, Index Ventures, Bain Capital Ventures, and Firstminute.
What We're Looking For:
- An experienced Senior Research Engineer who has built, trained, and optimized modern transformer-based language models.
- Deep experience with PyTorch, distributed training, and turning cutting-edge research into production systems.
- Comfortable working in fast-moving, ambiguous environments with strong communication and collaboration skills.
Benefits and Perks:
- Comprehensive health coverage including medical, dental, and vision, plus 401(k) matching and equity.
- Unlimited PTO, company-wide winter break, and flexible work arrangements.
- Paid parental leave, professional development allowance, wellness stipends, and a sabbatical program after four years.
Lightning AI
Lightning AI builds an end-to-end platform for developing, training, and deploying AI systems, founded in 2019. They are a global company with offices in New York, San Francisco, Seattle, and London, backed by major venture capital firms, and foster a builder culture that values urgency, ownership, and open communication.