Skip to main content
Atria Health

Senior AI Scientist

RemoteUnited States only
Published
Role
AI / ML
Experience
Senior
Employment
Full-time
$180k–$260k/yr
Check eligibility

Open to US only. Set where you work from to check your eligibility.

No BS summary

Senior AI Scientist needed to develop medical models for clinical AI agenda, focusing on domain-specific and foundation models for preventive medicine. Requires 4+ years of hands-on experience training and fine-tuning deep learning models, strong Python/PyTorch/Hugging Face skills, and a genuine interest in healthcare.

Core skills

medical modelsfoundation modelsfine-tuning

Required skills

PythonPyTorchHugging Face ecosystemtransformersdatasetsPEFTTRLaccelerate

Optional skills

multimodal medical modelsclinical textimagingradiologypathologyophthalmologystructured labsphysiological signals

What you'll do

  • Own the medical modeling roadmap at Atria: identify the model families and training approaches most likely to win on our problems, and build the models that prove it.
  • Survey the open-source landscape (general-purpose and clinical/biomedical foundation models, vision and multimodal backbones, specialized medical models) and make defensible recommendations on what to fine-tune, distill, or build on.
  • Design and validate novel architectures where existing approaches fall short, particularly for longitudinal multi-modal data and the foundation-model direction.
  • Lead fine-tuning and post-training work: SFT, LoRA/QLoRA and other PEFT methods, DPO and related preference-tuning approaches, RLHF/RLAIF where appropriate, continued pre-training, and distillation.
  • Curate high-quality training datasets from Atria's clinical data: sampling, labeling strategy, deduplication, contamination checks, train/test hygiene, and PHI-safe handling throughout.
  • Design and run rigorous evaluation: held-out clinical benchmarks, comparison against frontier closed-source baselines, calibration analysis, subgroup performance, and the ablations that tell you which design choices actually earned their keep.
  • Run training and evaluation on appropriate infrastructure (single-node and multi-GPU), with proper experiment tracking, reproducible configs, and the kind of logging that lets your future self understand what your past self did.
  • Stay current with the literature: training methods, fine-tuning techniques, medical foundation models, multimodal architectures.
  • Collaborate with clinicians on what "good" looks like for each model, and with research partners on joint training and evaluation work.

What they require

  • A bias toward shipping models, not papers about models. You would rather have a working v1 in front of clinicians next month than a beautiful methodology that ships next year.
  • A scrappy streak. You can pick up an unfamiliar fine-tuning technique, training framework, or clinical concept on a Wednesday and have a credible experiment running by Friday.
  • A serious drive to keep getting better. You read other people's code, papers, training logs, and post-mortems. You treat being wrong as cheap information.
  • Graduate degree (PhD preferred, Master's with strong research record) in computer science, machine learning, computational biology, biomedical informatics, or a closely related field — or a strong open-source track record in modern training and fine-tuning.
  • Hands-on experience training and fine-tuning modern deep learning models, with a track record of shipped or published models you personally trained: 4+ years.
  • Deep, current fluency with modern fine-tuning and post-training methods: SFT, PEFT (LoRA, QLoRA, adapters), preference tuning (DPO and successors), distillation, and continued pre-training.
  • Strong working knowledge of the open-source model ecosystem: which models are state of the art, which are overrated, and what's worth fine-tuning for a given problem.
  • Strong Python and PyTorch, with hands-on experience with the Hugging Face ecosystem (transformers, datasets, PEFT, TRL, accelerate) or equivalent training stacks.
  • Practical experience with training infrastructure: distributed training, mixed precision, efficient data loading, experiment tracking (W&B;, MLflow, or similar).
  • Discipline around evaluation and ablation: you treat benchmarking, calibration, and "what would this have looked like without that change?" as part of the modeling work.
  • Genuine interest in healthcare and the responsibility that comes with building models that affect patient care.

Benefits

  • Excellent health and wellness benefits, fully covered by Atria, effective date of hire
  • OneMedical membership for employees & dependents, giving access to 24/7 virtual care
  • Fertility & family planning
  • Company-covered preventive health screenings through partner hospitals (calcium score)
  • Fitness Perks, including Wellhub
  • + 401k contributions and 4% match starting after 6 months
  • Flexible Time Off
  • Continuing medical education (CME) and CEU support for professional licensure

The Atria Health Institute is a membership-based primary and specialty health care practice with a focus on prevention and longevity. We bring together a multidisciplinary team of renowned physicians to provide proactive, preventive, and precision-based care for Atria members and their families.

Healthcare
$180k–$260k/yr