Machine Learning Engineer - Large Language Models

📋 Description

• Fine-tune and optimize Large Language Models for healthcare apps using SFT/PEFT/DPO/PPO.
• Develop and enhance Retrieval-Augmented Generation (RAG) pipelines for AI retrieval.
• Collect, clean, and curate datasets to train and evaluate models.
• Convert, optimize, and package models for production across serving environments.
• Collaborate with cross-functional teams to deliver robust DL solutions for healthcare.
• Stay updated on LLM advances to improve model performance and reliability.

🎯 Requirements

• At least 5 years of professional experience in production-grade software and deep learning.
• Degree in CS, data science, ML, or related field; Master's/PhD preferred.
• Proven experience with Large Language Models, including fine-tuning and optimization techniques.
• Hands-on expertise with Axolotl and Transformer-based architectures.
• Experience with PyTorch.
• Familiarity with model serving tech like vLLM, Text Generation Inference, and llama.cpp.

🎁 Benefits

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