Back to Jobs

Machine Learning Engineer - Fine Tuning

Baseten San Francisco, New York
Posted 22 hours, 34 minutes ago
Deadline: Not specified
Full Time Senior Artificial Intelligence

As a Machine Learning Engineer specializing in Fine-Tuning at Baseten, you'll create exceptional value for customers by leveraging our world-class infrastructure to fine-tune large language models and/or other modalities, working directly with customers to achieve their specific goals. You'll build scalable pipelines, implement parameter-efficient techniques, and ensure a seamless transition to inference. This customer-facing role requires both technical expertise in foundation model adaptation and the ability to translate customer needs into effective solutions. You'll also help shape our product roadmap by identifying common patterns in customer requirements and working with product teams to develop reusable components and features , reducing the need for custom services and streamlining the fine-tuning process for everyone.

Requirements

REQUIREMENTS

Bachelor’s degree in Computer Science, Engineering, or related field

3+ years of experience in ML engineering with focus on model training and fine-tuning

Experience with advanced fine-tuning frameworks such as Axolotl, Unsloth, Transformers, TRL, PyTorch Lightning, or Torch Tune, enabling efficient model adaptation and optimization

Hands-on experience fine-tuning or pre-training LLMs or other foundation models

Excellent communication skills for explaining complex concepts to varied audiences

NICE TO HAVE

Experience working with customers to deliver technical solutions

Track record of delivering ML projects to enterprise customers

Knowledge of distributed training systems and efficiency optimization techniques

Experience with advanced alignment and adaptation techniques including RLHF, DPO, constitutional AI, prompt tuning, reinforcement learning with execution feedback, PPO, or other emerging alignment methods

Knowledge of prompt engineering and domain adaptation methods

Contributions to open-source fine-tuning projects or tools

Experience building user-friendly interfaces for fine-tuning workflows

Experience with cloud platforms (AWS, GCP, Azure) and containerization technologies

Benefits

Competitive compensation package.

This is a unique opportunity to be part of a rapidly growing startup in one of the most exciting engineering fields of our era.

An inclusive and supportive work culture that fosters learning and growth.

Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.

Salary
$150000 - $225000
per yearly
Company Size
Employment Type
Full Time
Work Mode
On-site (San Francisco, New York)
Apply Externally
Notice: You are about to leave RemoteWok and apply on an external site.
The application process will continue on the employer's website.
View Company Profile

Location

San Francisco, New York