Staff ML Engineer

AnaVation LLC; Remote; GBP 90,000 - GBP 130,000; Full time

Job summary

A promising high-tech startup is looking for a Staff AI Engineer to lead design efforts on their AI-driven mental health platform. The ideal candidate will have 8+ years of experience in Machine Learning, significant expertise in agentic orchestration, and a strong technical background in Python and SQL. This role offers a creative work environment, opportunities for career development, and the flexibility to work remotely or in a hybrid setup with unlimited vacation. Join a talented team to innovate healthcare solutions.

Benefits

Unlimited vacation

Health and well-being program

Career development opportunities

Qualifications

• 8+ years of professional experience in Machine Learning with complex AI products.
• Experience building production-grade AI systems and multi-model coordination.
• Practical experience with LLM fine-tuning and evaluation.

Responsibilities

• Lead the design of multi-agent workflows focusing on reliability and clinical safety.
• Collaborate with AI Research for post-training initiatives tailored for therapeutic domains.
• Define the long-term AI roadmap and oversee the development of robust evaluation frameworks.

Skills

ML Expertise

Agentic Orchestration

Applied Training Knowledge

Technical Stack

High-Level Leadership

Communication

Tools

Python

SQL

TypeScript

Job description

Overview

Sword Health is shifting healthcare from human-first to AI-first through its AI Care platform, making world-class healthcare available anytime, anywhere, while significantly reducing costs for payers, self-insured employers, national health systems, and other healthcare organizations. Sword began by reinventing pain care with AI at its core, and has since expanded into women’s health, movement health, and more recently mental health. Since 2020, more than 700,000 members across three continents have completed 10 million AI sessions, helping Sword\'s 1,000+ enterprise clients avoid over $1 billion in unnecessary healthcare costs. Backed by 42 clinical studies and over 44 patents, Sword Health has raised more than $500 million from leading investors including Khosla Ventures, General Catalyst, Transformation Capital, and Founders Fund.

As a Staff AI Engineer on the Mind team, you will be the technical cornerstone of our end-to-end mental health platform. We view Mind as the “Claude Code for mental health,” a highly sophisticated, multi-agent system that manages complete patient care pathways.

In this staff-level role, you will architect the next generation of our agentic infrastructure. You will move beyond implementation to design the orchestration layers, memory systems, and RAG frameworks that allow multiple LLMs to work in concert. While you will work closely with our AI Research team on model training today, you will play a pivotal role in evolving our applied training capabilities and specialized fine-tuning for clinical environments.

What you’ll be doing

• Agentic Architecture: Lead the design and scaling of complex multi-agent workflows, focusing on reliability, clinical safety, and sophisticated orchestration of multiple LLMs.
• Applied Model Evolution: Collaborate with AI Research to support and eventually spearhead post-training initiatives (SFT, RLHF) tailored for therapeutic domains.
• Technical Strategy: Define the long-term AI roadmap for the Mind team, balancing cutting-edge research with the pragmatic, high-availability needs of a production environment.
• Infrastructure Leadership: Oversee the development of robust evaluation frameworks and data pipelines that enable rapid iteration on both RAG performance and model behavior.
• Bridge Research & Production: Act as the primary technical liaison between AI Research and Engineering, ensuring state-of-the-art breakthroughs are successfully translated into scalable product features.

What you need to have

• ML Expertise: 8+ years of professional experience in Machine Learning, with a proven track record of shipping complex AI products at scale.
• Agentic Orchestration: Significant experience building and maintaining production-grade Agentic AI systems involving multi-model coordination, complex memory management, and tool-use.
• Applied Training Knowledge: Practical experience with LLM fine-tuning and evaluation. You understand the data requirements for post-training and are eager to drive these capabilities within an applied engineering context.
• Technical Stack: Expert-level Python and SQL (TypeScript is a plus); deep familiarity with modern AI orchestration frameworks and production software engineering best practices.
• High-Level Leadership: Experience setting technical direction, navigating deep ambiguity, and providing high-impact mentorship to senior engineers.
• Communication: Exceptional ability to translate technical complexity into strategic business value for both leadership and cross-functional stakeholders.

What we would love to see

• Experience in MedTech or regulated environments where AI safety and interpretability are paramount;
• Expertise in cloud-native AI infrastructure (GCP/Vertex AI or AWS/SageMaker);
• Active contributions to the LLM/Agentic AI open-source community;
• Previous AI research experience.

Benefits & environment

• A stimulating, fast-paced environment with lots of room for creativity;
• A bright future at a promising high-tech startup company;
• Career development and growth, with a competitive salary;
• The opportunity to work with a talented team and to add real value to an innovative solution with the potential to change the future of healthcare;
• A flexible environment where you can control your hours (remote) with unlimited vacation;
• Access to our health and well-being program (digital therapist sessions);
• Remote or Hybrid work policy.

Sword Health complies with applicable Federal and State civil rights laws and does not discriminate on the basis of Age, Ancestry, Color, Citizenship, Gender, Gender expression, Gender identity, Gender information, Marital status, Medical condition, National origin, Physical or mental disability, Pregnancy, Race, Religion, Caste, Sexual orientation, and Veteran status.