Senior Applied AI/ML Scientist

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 USDNew York, NY Salary Range$180,400—$225,500 USDDenver, CO Salary Range$165,600—$207,000 USDAll 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.