Staff Software Engineer, AI Model Lifecycle

Crusoe•Sunnyvale, CA

4d•$250,000 - $300,000

About The Position

The Staff Software Engineer for the AI Model Lifecycle team will help build a comprehensive managed platform for the application development lifecycle, with a focus on machine learning models, including large language models (LLMs). The role involves developing and operating scalable systems and pipelines for model training, fine-tuning, alignment, evaluation, and experiment management.

Requirements

Advanced degree in Computer Science, Engineering, or a related field.
4–5+ years of industry experience leading and driving impactful projects in AI.
Experience in generative AI, including large language models and multimodal systems.
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 Haves

Proficiency in Go or Python for large-scale, production-level services, and PyTorch.
Contributions to open-source AI projects such as vLLM or similar frameworks.
Experience optimizing performance on GPU systems and inference frameworks.

Responsibilities

Manage fine-tuning systems for large foundation models, including supervised fine-tuning (SFT), parameter-efficient fine-tuning (PEFT), LoRA, and adapters.
Build multi-node orchestration, checkpointing, failure recovery, and cost-efficient scaling for fine-tuning systems.
Implement and maintain end-to-end LLM training pipelines.
Develop reinforcement fine-tuning and reinforcement learning capabilities for fine-tuning and training.
Build distillation and reinforcement learning pipelines, including preference optimization, policy optimization, and reward modeling.
Support dataset, model, and experiment management, including versioning, lineage, evaluation, and reproducible fine-tuning at scale.