Senior Staff Software Engineer AI Model Lifecycle
You will build a managed platform for the AI application development lifecycle. You will manage fine-tuning systems and training pipelines for large foundation models, support multi-node orchestration and failure recovery, develop reinforcement learning and distillation workflows, and manage datasets, models, experiments, evaluations, and lineage.
Responsibilities
- Manage fine-tuning systems for foundation models
- Orchestrate multi-node training and checkpointing
- Implement end-to-end LLM training pipelines
- Develop reinforcement learning and reinforcement fine-tuning workflows
- Build distillation and preference or policy optimization pipelines
- Manage dataset, model, and experiment versioning
- Maintain lineage and evaluation systems
- Enable reproducible fine-tuning at scale
Requirements
- Advanced degree in Computer Science, Engineering, or a related field
- 8+ years of industry experience leading impactful AI projects
- Experience in Generative AI, large language models, and multimodal models
- Hands-on experience training, fine-tuning, and aligning LLMs with reinforcement learning and reinforcement fine-tuning
- Proficiency in Golang or Python
- Proficiency with PyTorch
Benefits
- Restricted Stock Units
- Paid time off
- Paid holidays
- Comprehensive health insurance
- Dental insurance
- Vision insurance
- Employer HSA contributions
- Paid parental leave
- Paid life insurance
- Short-term disability insurance
- Long-term disability insurance
- Professional development
- Tuition reimbursement
- Mental health and wellness support
- Commuter benefits
- Cell phone stipend
- 401(k) retirement plan with company match up to 4% of salary
- Volunteer time off