Staff AI/ML Software Engineer, Model Distillation & Fine-Tuning

Bachelor's degree in Computer Science, Machine Learning, Data Science, Mathematics, or equivalent practical experience.

8+ years of software engineering or applied ML research experience, including work where you set the technical direction others built against, made the architectural calls on an ML system, and brought other engineers along with you.

Deep proficiency in PyTorch.

Hands-on experience fine-tuning large language models or vision-language models, with results you can speak to in detail.

Practical experience with at least two of: knowledge distillation, parameter-efficient fine-tuning, pruning, or quantization.

Based in or willing to work hybrid out of Mountain View, CA or Seattle, WA, reporting to the office three days per week at minimum.

What Can Give You a Competitive Advantage (Preferred Qualifications)

Master's degree or Ph.D. in Computer Science, Artificial Intelligence, or a related field.

shipping a quantized model to a specific hardware target, including working through the accuracy regressions that surfaced along the way.

Familiarity with the broader training ecosystem (Hugging Face, DeepSpeed, Ray, or Megatron) and experience managing dataset pipelines at scale.

Domain experience in conversational AI, human-computer interaction, smart spaces, or deploying multimodal models in consumer-facing products.

Open-source contributions to foundation model tuning libraries, or published research on model compression, distillation, or efficient AI.

Ability to communicate complex AI training concepts and architectural trade-offs to cross-functional product and engineering teams.

The compensation information is a good faith estimate only.

It is based on what a successful applicant might be paid in accordance with applicable state laws.

The compensation may not be representative for positions located outside of the California Bay Area.

The salary range for this role is ($189,300 - $290,700).

The actual base salary a successful candidate will be offered within this range will vary based on factors relevant to the position.

Bonus Potential: An incentive pay program offers payouts based on company performance, job level, and individual performance.

Company Vehicle: Upon successful completion of a motor vehicle report review, you will be eligible to participate in a company vehicle evaluation program, through which you will be assigned a General Motors vehicle to drive and evaluate.

Note: program participants are required to purchase/lease a qualifying GM vehicle every four years unless one of a limited number of exceptions applies.

This Job may be eligible for relocation benefits.

#LI-SA2