Staff AI Researcher (Healthcare)

TL;DR

Staff AI Researcher (Healthcare): Develop AI solutions using extensive medical data sets to improve health outcomes with an accent on fine-tuning LLMs and generative models for healthcare domains. Focus on building prototypes, redesigning data pipelines, and developing evaluation metrics for model performance.

Location: Remote, United States

Company

Empower primary care physicians with technology to keep patients healthy and prevent unnecessary hospitalizations using one of the largest medical data sets.

What you will do

• Build prototypes using off-the-shelf and novel AI techniques for optimization.
• Work with large, complex medical data sets including records, diagnoses, claims, and prescriptions.
• Redesign pipelines and systems for growing data and query needs.
• Fine-tune and adapt pre-trained generative models to healthcare tasks.
• Develop evaluation metrics and benchmarks for AI/ML models.
• Design feature engineering pipelines for data processing and model optimization.
• Set engineering standards including code quality, testing, and release processes.
• Deliver POC solutions balancing speed, scalability, and time-to-market.

Requirements

• BS/BTech or higher in Computer Science or related field
• 3+ years deep learning and LLM experience
• 8+ years machine learning and statistical analysis experience
• 3+ years Python experience
• Experience with incomplete, unrepresentative, or mislabeled data challenges
• Experience with large-scale distributed systems (e.g., Spark)
• 3+ years proficiency in selecting tools for data optimization problems

Nice to have

• Ph.D. or Master's in quantitative discipline (CS with AI/ML, Statistics, etc.)
• Proficiency communicating analysis to non-experts
• Experience with security for sensitive data
• Experience with Databricks/MLflow
• Production-ready agentic systems
• Deep learning frameworks (PyTorch, TensorFlow, Keras)
• Leadership, self-direction, publications (NeurIPS, ICML, etc.), competition wins
• Health-tech knowledge (EHR, clinical data)

Culture & Benefits

• Partner with engineering and analytics teams to integrate AI into products.
• Access to extensive medical data from millions of patients.
• Focus on high-quality engineering processes.