Machine Learning Engineer - Large Language Models

📋 Description

• Fine-tune and optimize LLMs for healthcare and life sciences using SFT, PEFT, DPO, PPO.
• Develop and enhance RAG pipelines to improve AI-powered information retrieval.
• Collect, prepare, clean, and curate high-quality datasets for training and evaluation.
• Convert and deploy AI models for production across serving environments with scalability.
• Collaborate with cross-functional teams to build and deploy DL solutions for healthcare.
• Stay updated on LLMs, ML infra, and AI best practices to boost performance.

🎯 Requirements

• 5+ years of experience in production-grade software and deep learning.
• Degree in CS, Data Science, ML, or related field; MS/PhD preferred.
• Experience with Large Language Models, including fine-tuning and optimization.
• Hands-on with Axolotl and Transformer-based architectures.
• Experience with PyTorch.
• Familiarity with model serving tools such as vLLM, TGI, and llama.cpp.
• Understanding of model quantization and optimization for constrained resources.
• Strong problem-solving and collaboration in distributed teams.
• Passion for AI innovation and real-world ML applications.

🎁 Benefits

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