AI Research Scientist - Remote

Overview

This role focuses on advancing healthcare reimbursement through innovative AI research and development, combining technical expertise with clinical insights.

In Short

• Own the architecture, training, and optimization of large-scale transformer-based pipelines.
• Design and iterate on new transformer and hybrid text architectures.
• Drive the research roadmap and publish meaningful advances.
• Build retrieval-augmented generation (RAG) pipelines.
• Convert research prototypes into reliable services.
• Collaborate with product and design teams.
• 3+ years of experience in training large transformer models.
• Fluency with relevant tools like PyTorch and Docker.
• Expertise in text modeling is required.
• Bonus points for familiarity with healthcare ontologies.

Requirements

• 3+ years training large transformer models in Python/PyTorch at scale.
• Peer-reviewed publications or significant open-source work in text modelling.
• Proven end-to-end ownership from architecture to deployment.
• Fluency with Lightning/FSDP, Pytorch, Hugging Face, WandB, Ray/Kubeflow, Docker.
• Deep expertise in text modelling.

Benefits

• Top-of-market compensation (salary + equity).
• Flexible PTO & hybrid culture.
• Mission-driven, collaborative team.
• Twice-yearly offsites to align and celebrate.