6 months ago
Base Salary
$220k - $320k/yr
Responsibilities
- Lead projects from data intake through the full training pipeline.
- Build and maintain data processing pipelines for training data.
- Create dashboards and visualization tools for training metrics.
- Train models using internal frameworks and iterate based on evaluations.
- Develop benchmarks and evaluation frameworks for model performance.
- Automate portions of the training workflow to improve consistency.
- Ship research features into production settings.
- Apply techniques in SFT, RL, and model optimization.
- Collaborate with infrastructure engineers to scale training.
- Understand customer use cases to inform training strategies.
Requirements
- 2+ years of experience training AI models using PyTorch.
- Hands-on experience with post-training LLMs using SFT or RL.
- Strong understanding of transformer architectures.
- Experience with LLM-specific training frameworks.
- Experience training on NVIDIA GPUs.
- Strong data processing skills and ETL pipeline experience.
- Track record of creating benchmarks and evaluations.
- Ability to apply research techniques to production systems.
Benefits
- Competitive compensation and equity in a high-growth startup.
- Comprehensive benefits package.