Heidi · 1 month ago
Senior AI Model Engineer
San Francisco, CA
Full-time
Hybrid
Senior LevelHeidi is building an AI Care Partner that supports clinicians every step of the way, from documentation to delivery of care. The Senior AI Model Engineer will be a technical leader responsible for solving critical model-related challenges and driving breakthroughs in model capabilities.
Artificial Intelligence (AI)Business Process Automation (BPA)Health CareSoftware
Responsibilities
Lead Model R&D Initiatives: Lead major research and development projects focused on step-changes in model performance, from ideation and experiment design to productionisation
Solve the Hardest Problems: Tackle the most ambiguous and challenging technical problems related to model behaviour, alignment, and safety
Build cutting-edge fine-tuned LLMs within Heidi using the latest tools and frameworks
Partner with product, medical knowledge and design teams to automate prompt engineering, build agentic AI, create model evaluation frameworks
Develop Novel Techniques: Go beyond existing literature to develop novel fine-tuning or data generation techniques tailored to our unique healthcare domain
Mentor and Guide: Mentor mid-level and junior engineers, sharing your deep expertise and fostering a culture of scientific rigor and innovation
Qualification
Required
You've a proven track record of leading research or development projects that have resulted in significant improvements to large-scale AI models
You've previous hands-on experience building scalable AI products or features
You've shipped AI and LLM-powered features to real users in a production environment
You are considered an expert in Python, deep learning frameworks (PyTorch/TensorFlow), and the underlying theory of transformer architectures
You have a world-class intuition for AI/ML, built on a deep and solid foundation in mathematics and statistics
You have the ability to translate vague product needs into concrete research questions and well-designed experiments
A PhD in Computer Science, AI, Statistics, or a related field is highly preferred
Preferred
A strong publication record in top-tier AI conferences
Experience building and leading a research-focused team
Expertise in reinforcement learning or data-centric AI
Benefits
Flexible hybrid working environment, with 3 days in the office.
Additional paid day off for your birthday and wellness days
Special corporate rates at Anytime Fitness in Melbourne, Sydney tbc.
A generous personal development budget of $500 per annum
Become an owner, with shares (equity) in the company, if Heidi wins, we all win
Company
Heidi
Heidi is the AI Care Partner designed to expand clinical capacity by automating administrative work – documentation, form filling, and task management - so clinicians can focus on patients.
Founded in 2019
51-200 employees
H1B Sponsorship
Heidi has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2024 (1)
Funding
Current Stage
Growth StageTotal Funding
$91.55MKey Investors
Point72 VenturesGrow Digital Health Midlands AcceleratorHeadline
2025-10-06Series B· $65M
2025-05-13Non Equity Assistance
2025-03-03Series A· $16.6M
Leadership Team
Thomas Kelly
Co-Founder & CEO
Waleed Mussa
Co-Founder
Recent News
2025-12-26
Startup Daily
2025-12-24
HealthCareIT News
2025-12-10
Company data provided by crunchbase