Senior Staff Software Engineer, AI Model Lifecycle

FDE Times; San Francisco (CA); On-site; USD 180,000 - 260,000; Full time

Benefits offered by this job

Health Insurance

Paid Time Off

401(k) Retirement Plan

Job summary

FDE Times is seeking an experienced AI Infrastructure Engineer to build and manage scalable training and fine-tuning systems for foundation models. You will lead distributed orchestration, checkpointing, and cost-efficient scaling while advancing reinforcement learning and tuning pipelines.

Ideal candidates have a strong background in LLMs, generative AI, and distributed training, with hands-on experience in PyTorch, Python, and ML tooling.

Qualifications

• Advanced degree in computer science, engineering, or related field.
• 4+ years of industry experience leading AI projects.
• Experience with generative AI, training and fine-tuning LLMs, and RL techniques.

Responsibilities

• 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, and evaluation pipelines for reproducible training at scale.

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

Education

Advanced degree in CS/Engineering or related field

Tools

PyTorch

Python

Go

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