Senior Staff Software Engineer, AI Model Lifecycle

Job Description

Build and manage scalable fine-tuning and end-to-end training systems for foundation models, including distributed orchestration, checkpointing, failure recovery, and cost-efficient scaling. Develop reinforcement learning, reinforcement fine-tuning, distillation, dataset and model management, and evaluation pipelines that support reproducible training at scale.

Requirements: The role requires an advanced degree in computer science, engineering, or a related field, along with 4–5 or more years of industry experience leading impactful AI projects. Candidates should have generative AI experience and hands-on expertise training, fine-tuning, and aligning LLMs with reinforcement learning and reinforcement fine-tuning techniques; Python or Go, PyTorch, open-source AI contributions, and GPU optimization are bonuses.

Key Skills: Large Language Models, Generative AI, Multimodal AI, Reinforcement Learning, Reinforcement Fine-Tuning, LLM Fine-Tuning, Supervised Fine-Tuning, Parameter-Efficient Fine-Tuning, LoRA, PyTorch, Python, Go, Distributed Training, GPU Performance Optimization, Model Evaluation, Experiment Management

Benefits: Restricted Stock Units, Paid Time Off, Paid Holidays, Health Insurance, Dental Insurance, Vision Insurance, Health Savings Account Contributions, Paid Parental Leave, 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, Company Match, Volunteer Time Off, Bonus