Machine Learning Engineer, Baseten

• If you’re passionate about tackling impactful challenges and building transformative solutions from the ground up, we invite you to join us on this exciting journey!
• This customer-facing role requires both technical expertise in foundation model adaptation and the ability to translate customer needs into effective solutions
• 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
• If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you

Desirable

• Join our dynamic team at Baseten, where we’re revolutionizing AI deployment with cutting-edge inference infrastructure
• 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
• 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
• Design comprehensive fine-tuning strategies that translate customer requirements into effective technical approaches—finding the optimal combination of data preparation, training techniques, and evaluation methods to deliver solutions that precisely address customer needs
• Develop tools to enable non-ML experts to fine-tune models effectively
• Design and implement scalable fine-tuning pipelines for large language models and other AI modalities
• Work directly with customers to understand requirements and guide technical implementation
• Serve as the technical point of contact for customers throughout their fine-tuning journey
• Utilize state-of-the-art parameter-efficient fine-tuning methods (LoRA, QLoRA)
• Build systems for efficient data preparation, evaluation, and deployment of fine-tuned models
• Research and apply cutting-edge techniques in instruction tuning and model customization
• Create frameworks to evaluate fine-tuned model performance against base models
• Implement best-in-class distributed training techniques like FSDP and DDP across various hardware configurations