Senior Research Engineer, LLM Training & Post-Training
Summary
Senior Research Engineer building and optimizing PyTorch-based LLM training pipelines, post-training methods, and distributed training systems for large language models.
- Build and improve PyTorch based training infrastructure and tooling
- Collaborate with customers to translate real world workloads into model improvements
- Contribute to open source projects with features tooling improvements and documentation
- Design build and optimize training pipelines for large language models
- Design evaluation methodologies and benchmark models and analyze failure modes
- Improve model quality using supervised fine tuning continued pretraining preference optimization reinforcement learning and evaluation
- Investigate model training issues convergence instability communication overhead and performance bottlenecks
- Optimize distributed training across multi GPU environments
- Partner with research and engineering teams to build production ready AI systems
Perks/Benefits:
- 401k matching
- Company holidays
- Comprehensive health coverage
- Dental coverage
- Flexible time off
- Floating holidays
- Hybrid work model
- In-office meals
- Paid parental and family leave
- Pension contributions
- Professional development allowance
- Sabbatical program
- Two weeks winter break
- Vision coverage
- Wellness benefits
- Work from home stipend