Judi Health is looking to infuse cutting-edge AI/ML/GenAI into healthcare by developing Large Language Models (LLMs) to advance state-of-the-art AI-based automation for Healthcare, aiming to build Conversational AI products that increase productivity and create enhanced customer experiences.
Requirements
- 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.
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.
Other
- Minimum 4 years of experience in industry with a strong focus on applied AI/ML research and development
- 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.
