Remote Machine Learning Engineer - Large Language Models

Description:

• John Snow Labs is seeking a highly skilled and experienced Machine Learning Engineer to join their 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 that includes professionals from software engineering, data science, and medicine.
• Key responsibilities include adapting 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).
• The engineer will optimize LLMs for Retriever-Augmented Generation (RAG) to enhance decision-making and information retrieval capabilities.
• Responsibilities also include collecting, cleaning, and refining healthcare datasets for training LLMs to ensure high-quality data provisioning.
• The role requires converting models into various formats suitable for production environments, ensuring their readiness for real-world application.

Requirements:

• Candidates must have 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 is required, with a M.Sc. or Ph.D. degree strongly preferred.
• Demonstrated expertise in model tuning frameworks like Axolotl is necessary.
• Familiarity with model serving frameworks, including vLLM, TGI, and llama-cpp, is required 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 is essential.
• Hands-on experience with Transformer architectures and proficiency in machine learning frameworks such as PyTorch is required.
• Applicants must include a cover letter that mentions ‘John Snow Labs’ and explains their academic, professional ML engineering, and recent LLM engineering experience.

Benefits:

• John Snow Labs offers a fully virtual company environment, collaborating across 28 countries.
• The position includes a competitive package and compensation plan.
• Employees benefit from being part of an industry leader and respected brand name.
• The company provides opportunities for learning and development.