Crusoe
Senior Staff Software Engineer, AI Model Lifecycle
San Francisco, CA - US. Posted 8 days ago
The role
This person will build a managed platform for the application development lifecycle focusing on Machine Learning models and Large Language Models. They will manage fine-tuning systems, training pipelines, and experiment management.
What you will do
- Manage fine-tuning systems for large foundation models 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.
- Dataset, model, and experiment management including versioning, lineage, evaluation, and reproducible fine-tuning at scale.
Required qualifications
- Advanced degree in Computer Science, Engineering, or a related field.
- 4-5+ 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
- 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
About the company
Crusoe is a vertically integrated AI infrastructure company that owns and operates each layer of the stack to power AI workloads. They use an energy-first approach to solve AI compute and power bottlenecks.