Machine Learning Engineer - Large Language Models

John Snow Labs; United States; Remote; USD 41,328 - 82,656; Full time

Benefits offered by this job

Fully virtual collaboration across 28 countries

Competitive package and compensation plan

Learning and development opportunities

Job summary

John Snow Labs, a leader in AI and NLP, is looking for a Machine Learning Engineer to join its fully virtual team. This role involves advancing healthcare AI through innovative model training and deployment strategies in a highly collaborative environment.

The ideal candidate will have over 5 years of experience, a strong academic background, and a proven track record in deep learning. John Snow Labs offers competitive compensation and opportunities for professional development.

Qualifications

• 5+ years of professional experience in software engineering and building production-grade deep learning solutions.
• Strong background in machine learning and natural language processing.
• Experience with model quantization techniques.

Responsibilities

• Adapt LLMs to diverse healthcare use-cases using various techniques.
• Optimize LLMs for Retriever-Augmented Generation to enhance capabilities.
• Collect, clean, and refine datasets for training LLMs.
• Convert models into production-compatible formats.

Skills

Machine Learning

Natural Language Processing (NLP)

Large Language Models (LLMs)

Software Engineering

Deep Learning Solutions

Model Tuning Frameworks

Transformer Architectures

Education

M.Sc. or Ph.D. in Computer Science or Data Science

Tools

Axolotl

vLLM

TGI

llama-cpp

PyTorch

Job description

• Contract
• Compensation: USD 30 - USD 60 per hour

Company Description

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. Founded in 2015, it helps healthcare and life science companies build, deploy, and operate AI products and services. John Snow Labs is the winner of the 2018 AI Solution Provider of the Year Award, the 2019 AI Platform of the Year Award, the 2019 International Data Science Foundation Technology award, and the 2020 AI Excellence Award.

John Snow Labs is the developer of Spark NLP – the world’s most widely used NLP library in the enterprise – and is the world’s leading provider of state-of-the-art clinical NLP software, powering some of the world’s largest healthcare & pharma companies. John Snow Labs is a global team of specialists, of which 33% hold a Ph.D. or M.D. and 75% hold at least a Master’s degree in disciplines covering data science, medicine, software engineering, pharmacy, DevOps and SecOps.

Job Description

We are seeking a highly skilled and experienced Machine Learning Engineer to join our team. The ideal candidate will have a robust background in machine learning, natural language processing (NLP), and Large Language Models (LLMs). The role focuses on training, tuning, and evaluating AI models as part of a diverse team which includes professionals from software engineering, data science, and medicine. This role offers an opportunity to be at the forefront of technological advancement in AI and healthcare, contributing to innovations that advance the healthcare, life science, and open‑source AI communities.

Responsibilities

• Adapt LLMs to diverse healthcare use‑cases using techniques such as Sparse Fine‑Tuning (SFT), Prompt Engineering Fine‑Tuning (PEFT), Direct Parameter Optimization (DPO), and Proximal Policy Optimization (PPO).
• Optimize LLMs for Retriever‑Augmented Generation (RAG) to enhance decision‑making and information retrieval capabilities.
• Collect, clean, and refine healthcare datasets for training LLMs to ensure high‑quality data provisioning.
• Convert models into various formats suitable for production environments, ensuring their readiness for real‑world application.

Requirements

• 5+ years of hands‑on professional experience in software engineering, building production‑grade deep learning solutions.
• An academic degree in computer science, data science, or a related degree. M.Sc. or Ph.D. degree is strongly preferred.
• Demonstrated expertise in model tuning frameworks like Axolotl.
• Familiarity with model serving frameworks, including vLLM, TGI, and llama‑cpp, to support the deployment and scalability of machine learning models.
• Knowledge of model quantization techniques and frameworks to optimize AI models for performance in resource‑constrained environments.
• Hands‑on experience with Transformer architectures and proficiency in machine learning frameworks such as PyTorch.

Benefits

What John Snow Labs offers:

• A fully virtual company, collaborating across 28 countries
• Competitive package and compensation plan
• Industry leader and respected brand name
• Learning and development

Equal Employment Opportunity

We are proud to foster a workplace free from discrimination. Diversity of experience, perspectives, and background create a better work environment and better products. Whatever your identity we will give your application fair consideration.