Machine Learning Engineer - LLMs & Generative AI

📋 Description

• Lead development, training, and deployment of LLMs for clinical/biomedical domains.
• Apply SFT, RL-based methods (RLHF/RLVR), PEFT, prompt tuning, and RAG.
• Collaborate with researchers, clinicians, and engineers to design ML healthcare solutions.
• Build scalable infra for distributed training on TPU/GPU clusters.
• Design and evaluate models for robustness, bias, factuality, and explainability in healthcare.
• Stay current with AI research and contribute to publications and open-source initiatives.

🎯 Requirements

• 6+ years of software engineering or ML experience (3+ years with a PhD).
• Experience designing/training LLMs or large-scale generative models (GPT, PaLM, LLaMA, Claude, Gemma).
• Deep expertise in NLP, sequence modeling, and transformer architectures.
• Proficient in Python and ML libraries such as PyTorch or TensorFlow; strong ML pipelines.
• Experience with RL-based fine-tuning (e.g., RLHF) and evaluation of generative systems.
• Proven ability to lead technical projects and collaborate across teams.
• Bachelor's degree in Computer Science, Engineering, or a related technical field.

🎁 Benefits

• Interesting and meaningful work for every career stage.
• Great benefits package.
• Comprehensive medical, dental and vision insurance.
• 401K plan.
• Professional development and training opportunities for continuous learning.
• Work-life balance with flexible hours and PTO.