Senior Applied AI/ML Scientist

About Judi Health Judi Health is an enterprise health technology company providing a comprehensive suite of solutions for employers and health plans, including: Capital Rx, a public benefit corporation delivering full-service pharmacy benefit management (PBM) solutions to self-insured employers, Judi Health™, which offers full-service health benefit management solutions to employers, TPAs, and health plans, and Judi®, the industry’s leading proprietary Enterprise Health Platform (EHP), which consolidates all claim administration-related workflows in one scalable, secure platform. Together with our clients, we’re rebuilding trust in healthcare in the U.S. and deploying the infrastructure we need for the care we deserve. 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. Required Qualifications: Bachelor’s or Master’s degree in Computer Science, Machine Learning, or a related quantitative field. PhD in AI, Machine Learning or Computer Science preferred 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 range represents the low and high end of the anticipated base salary range. The actual base salary will depend on several factors such as: experience, knowledge, skills, and location of the job. Remote, US Salary Range $142,800—$225,500 USD New York, NY Salary Range $180,400—$225,500 USD Denver, CO Salary Range $165,600—$207,000 USD All employees are responsible for adherence to the Capital Rx Code of Conduct including the reporting of non-compliance. 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. By submitting an application, you agree to the retention of your personal data for consideration for a future position at Judi Health. More details about Judi Health's privacy practices can be found at https://www.judi.health/legal/privacy-policy.