Senior AI/ML Applied Scientist

About Us:

Judi Health is a health technology company offering a wide range of benefits administration solutions for employers and health plans. This includes Capital Rx, a public benefit corporation that provides full-service pharmacy benefit management (PBM) solutions to self-insured employers; Judi Health™, which offers comprehensive health benefit management solutions for employers, TPAs, and health plans; and Judi®, the industry’s leading proprietary Enterprise Health Platform. To learn more, visit www.judi.health.

Position Summary: 

Join our mission to infuse cutting-edge AI/ML/GenAI into healthcare as a Senior Applied AI/ML Scientist.  We are seeking Artificial Intelligence (AI) Applied Scientists to join our team and contribute to the development of Large Language Models (LLMs) aimed at advancing state-of-the-art for AI-based automation for Healthcare. As an Applied Scientist, you will play a crucial role in developing, improving, and exploring the capabilities of multimodal LLMs to solve complex real-world problems geared towards building Conversational AI products enabling increased productivity and magical experiences for our customers. 

The ideal candidate should be passionate about applying AI/ML concepts to difficult problems and develop scalable customized solutions. We want people who like working in a collaborative team environment and enjoy creating practical, efficient, and high-performance software that leverages Large Language Models (LLM), Multimodal Language Models(MLM), and other ML models and techniques to build amazing capabilities for our customers, partners, and employees. Most importantly, we are a mission-oriented, high-growth startup and we are looking for folks that are excited to be part of our journey to make lasting impact towards transforming healthcare. 

Responsibilities: 

• Investigate the latest LLM MLM models and evaluate and compare accuracy and efficiency for our in-domain use cases 

• Develop tools and processes to create data pipelines for ML training and customization 

• Finetune LLMs for the healthcare domain  

• Help develop and productionize machine learning (ML) solutions leveraging our customized models in the fields of Document understanding, Search and QnA, GenAI, Virtual Agents, etc. 

• Work across the complete lifecycle of ML model development, including problem definition, data exploration, feature engineering, model training, validation, and deployment. 

• Participate in code reviews, testing, and quality assurance processes. 

• Troubleshoot and resolve technical issues related to AI model, integration, deployment and backend services. 

• Document and communicate research and development processes, and architectural decisions. 

• Ensure the security and compliance of healthcare data, adhering to HIPAA regulations. 

• All employees are responsible for adherence to the Capital Rx Code of Conduct including reporting of noncompliance. 

Required Qualifications: 

• Bachelor’s or Master’s degree in Computer Science, Machine Learning, or a related quantitative field. 

• Minimum 4 years of experience in industry with a strong focus on applied AI/ML research and development 

• Strong grasp of OOP, Design Patterns, efficient algorithms, and quality software development. 

• Solid understanding of Machine Learning fundamentals such as Deep Neural Networks, Transformer-based LLM and MLMs, Boosted or standard decision trees, RAG, etc. 

• Strong proficiency in Python and familiarity with ML libraries such as PyTorch. 

• Experience with training (pre-training / fine-tuning / RLHF / post-training) autoregressive language models like LLaMA, GPT-x, Mistral etc. 

• Experience with open-source LLM platforms and training stack (e.g., HuggingFace, Deepspeed etc.). 

• Experience with ML platforms such as AWS Sagemaker, Azure ML, Databricks etc. 

• Excellent problem-solving skills, attention to detail, and a strong capacity for logical thinking. 

• Passion for customer, innovation, product quality 

• The ability to work collaboratively across multiple disciplines in an extremely fast-paced, startup environment. 

• Good written communication skills that enable collaboration in a remote environment. 

• Experience in healthcare or life sciences is a plus. 

This position description is designed to be flexible, allowing management the opportunity to assign or reassign duties and responsibilities as needed to best meet organizational goals.

Judi Health values a diverse workplace and celebrates the diversity that each employee brings to the table. We are proud to provide equal employment opportunities to all employees and applicants for employment and prohibit discrimination and harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, medical condition, genetic information, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state or local laws.