Machine Learning Engineer - Large Language Models

Responsibilities

• Adapt LLMs to diverse healthcare use-cases using techniques such as SFT, PEFT, DPO, and PPO
• Optimize LLMs for Retriever-Augmented Generation (RAG)
• Collect, clean, and refine healthcare datasets for training LLMs
• Convert models into various formats suitable for production environments

Requirements

• 5+ years of hands-on professional experience in software engineering, building production-grade deep learning solutions
• Academic degree in computer science, data science, or a related degree (M.Sc. or Ph.D. preferred)
• Demonstrated expertise in model tuning frameworks like Axolotl
• Familiarity with model serving frameworks, including vLLM, TGI, and llama-cpp
• Knowledge of model quantization techniques and frameworks
• Hands-on experience with Transformer architectures and proficiency in PyTorch

About the Company

John Snow Labs is an award-winning AI and NLP company, accelerating progress in data science by providing state-of-the-art software, data, and models. It helps healthcare and life science companies build, deploy, and operate AI products and services.