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
• Create 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

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