Machine Learning Engineer - Large Language Models

📋 Description

• Fine-tune and optimize Large Language Models for healthcare and life sciences.
• Develop and enhance Retrieval-Augmented Generation (RAG) pipelines.
• Collect, prepare, clean, and curate high-quality datasets for training and evaluation.
• Convert, optimize, and package AI models for production deployment across serving environments.
• Collaborate with cross-functional teams to deploy robust deep learning solutions for healthcare.
• Stay up to date with advancements in LLMs and ML infrastructure to improve performance.

🎯 Requirements

• At least 5 years of professional experience in production-grade software and DL.
• Degree in CS, Data Science, ML, or related field; MS/PhD preferred.
• Proven experience with Large Language Models, including fine-tuning.
• Hands-on experience with Axolotl and Transformer architectures.
• Experience with ML frameworks, particularly PyTorch.
• Familiarity with model serving tech such as vLLM, TGI, llama.cpp.

🎁 Benefits

• Fully remote work with an international team.
• Competitive compensation package.
• Opportunity to work on cutting-edge AI and LLM technologies.
• Continuous learning and professional development opportunities.
• Exposure to impactful projects in healthcare, life sciences, and open-source AI.
• Inclusive, diverse, and collaborative workplace with equal opportunity.