Machine Learning Engineer - Fine Tuning - Remote

Overview

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.

In Short

• Design comprehensive fine-tuning strategies for customer requirements.
• Develop tools for non-ML experts to fine-tune models.
• Implement scalable fine-tuning pipelines for AI modalities.
• Engage directly with customers to understand and guide technical implementations.
• Utilize state-of-the-art fine-tuning methods.
• Build efficient systems for data preparation and deployment.
• Research cutting-edge techniques in model customization.
• Create frameworks for evaluating model performance.
• Implement distributed training techniques across hardware configurations.
• Shape product roadmap with insights from customer interactions.

Requirements

• Bachelor’s degree in Computer Science, Engineering, or related field.
• 3+ years of experience in ML engineering with a focus on model training and fine-tuning.
• Experience with advanced fine-tuning frameworks like PyTorch Lightning.
• Hands-on experience fine-tuning or pre-training LLMs.
• Excellent communication skills.

Benefits

• Competitive compensation package including flexible PTO.
• Opportunity to be part of a rapidly growing startup.
• Inclusive and supportive work culture.
• Exposure to various ML startups for learning and networking.