Machine Learning Engineer - Fine Tuning

ABOUT BASETENBaseten provides the infrastructure, tooling, and expertise needed to bring great AI products to market - fast. Backed by top investors including IVP, Spark Capital, Greylock, and Conviction, we’re trusted by leading AI-driven innovators like Writer, Abridge, Bland, Patreon, Descript, Retool, and Zed to deliver industry-leading performance, security, and reliability for their mission-critical workloads. With our recent $75M Series C funding, we’re growing fast to make AI accessible across all products.THE ROLEAs 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.RESPONSIBILITIESDesign 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 needsDevelop tools to enable non-ML experts to fine-tune models effectivelyDesign and implement scalable fine-tuning pipelines for large language models and other AI modalitiesWork directly with customers to understand requirements and guide technical implementationServe as the technical point of contact for customers throughout their fine-tuning journeyUtilize state-of-the-art parameter-efficient fine-tuning methods (LoRA, QLoRA)Build systems for efficient data preparation, evaluation, and deployment of fine-tuned modelsResearch and apply cutting-edge techniques in instruction tuning and model customizationCreate frameworks to evaluate fine-tuned model performance against base modelsImplement best-in-class distributed training techniques like FSDP and DDP across various hardware configurationsREQUIREMENTSBachelor’s degree in Computer Science, Engineering, or related field3+ years of experience in ML engineering with focus on model training and fine-tuningExperience with advanced fine-tuning frameworks such as Axolotl, Unsloth, Transformers, TRL, PyTorch Lightning, or Torch Tune, enabling efficient model adaptation and optimizationHands-on experience fine-tuning or pre-training LLMs or other foundation modelsExcellent communication skills for explaining complex concepts to varied audiencesNICE TO HAVEExperience working with customers to deliver technical solutionsTrack record of delivering ML projects to enterprise customersKnowledge of distributed training systems and efficiency optimization techniquesExperience with advanced alignment and adaptation techniques including RLHF, DPO, constitutional AI, prompt tuning, reinforcement learning with execution feedback, PPO, or other emerging alignment methodsKnowledge of prompt engineering and domain adaptation methodsContributions to open-source fine-tuning projects or toolsExperience building user-friendly interfaces for fine-tuning workflowsExperience with cloud platforms (AWS, GCP, Azure) and containerization technologiesBENEFITSCompetitive compensation package (Flexible PTO, 401k, covered healthcare premiums).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.Apply now to embark on a rewarding journey in shaping the future of AI! 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.At Baseten, we are committed to fostering a diverse and inclusive workplace. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, or veteran status.

Apply Now