Senior Staff Software Engineer, AI Model Lifecycle

The Senior Staff Software Engineer for the AI Model Lifecycle team will play a crucial role in building a comprehensive managed platform for the entire application development lifecycle, with a specific focus on leveraging Machine Learning models, including Large Language Models (LLMs).

Responsibilities

Manage fine-tuning systems for large foundation models (SFT, PEFT, LoRA, adapters), including multi-node orchestration, checkpointing, failure recovery, and cost-efficient scaling; Implement and maintain end-to-end training pipelines for Large Language Models; RFT and Reinforcement learning to the fine tuning and training sections; Distillation and reinforcement learning pipelines (e.g., preference optimization, policy optimization, reward modeling); Dataset, model, and experiment management: versioning, lineage, evaluation, and reproducible fine-tuning at scale.

Requirements

Advanced degree in Computer Science, Engineering, or a related field; 8+ years of industry experience leading and driving impactful projects in the AI Space; Experience in Generative AI (Large Language Models, Multimodal); Hands-on experience training, fine-tuning, and aligning LLMs using Reinforcement Learning and Reinforcement Fine-Tuning (RFT) techniques; Proactive and collaborative approach with the ability to work autonomously; Passion for building cutting-edge AI products and solving challenging technical problems.

Nice to have

Proficiency in Golang or Python for large-scale, production-level services and PyTorch; Contributions to open-source AI projects such as vLLM or similar frameworks; Performance optimizations on GPU systems and inference frameworks.

Benefits

Competitive compensation; Bonus; Restricted Stock Units; Paid time off & paid holidays; Comprehensive health, dental & vision insurance; Employer contributions to HSA account; Paid parental leave; Paid life insurance, short-term and long-term disability; Professional development & tuition reimbursement; Mental health & wellness support; Commuter benefits (parking & transit); Cell phone stipend; 401(k) Retirement plan with company match up to 4% of salary; Volunteer time off