Senior AI/ML engineer

Reposted 4 Days Ago
Easy Apply
Be an Early Applicant
London, Greater London, England, GBR
In-Office
120K-150K Annually
Senior level
Fitness
The Role
The role involves developing, fine-tuning, and optimizing large language models, building ML infrastructure, and implementing CI/CD practices for AI/ML systems.
Summary Generated by Built In

500M+ downloads. 80M+ monthly users. A decade of building – and we’re still accelerating.

Flo is the world’s #1 health & fitness app worldwide on a mission to build a better future for female health. Backed by a $200M investment led by General Atlantic, we became the first product of our kind to reach a $1B valuation in 2024 – and we’re not slowing down.

With 7M paid subscribers and the highest-rated experience in the App Store’s health category, we’ve spent 10 years earning trust at scale. Now, we’re building the next generation of digital health – AI-powered, privacy-first, clinically backed – to help our users know their body better.

The job

We are looking for a Senior Software Engineer with deep expertise in AI/ML infrastructure to join our AI Platform team and help build the GenAI platform that powers every AI feature at Flo.

You will bridge core infrastructure, data engineering, and LLM development to deliver production-grade medical safety judges, fine-tuning pipelines, evaluation frameworks, and real-time personalisation. The team operates 60+ LLM-based evaluation judges, develops proprietary fine-tuned health models, and maintains active partnerships with Databricks, Google, OpenAI, Anthropic, and AWS.

What you’ll do

  • LLM Judge Ecosystem: build and scale Judge-as-a-Service, prompt registries, calibration pipelines, and evaluation orchestration using MLflow 3.x
  • Fine-Tuning and Serving: develop LoRA/SFT/preference optimisation pipelines for health-domain models (Llama, Gemma, MedGemma) and manage model serving at scale on Databricks
  • Data and Evaluation Pipelines: build synthetic Q&A generation, golden test sets, reward function engineering, and Delta table schemas in Unity Catalog for reliable, reproducible evaluation data
  • Infrastructure: maintain Terraform-managed AWS infrastructure (EKS, S3, IAM), Databricks AI Gateway, and CI/CD pipelines (GitHub Actions) with evaluation gates and progressive rollout
  • Cross-Functional Impact: collaborate with Product, Security, Analytics, and Medical teams, develop internal SDKs and APIs consumed by 5+ teams, and engage directly with technology partners on pre-release capabilities

Experience and skills

Must have:

  • Engineering maturity: 7+ years of software engineering, 4+ years focused on ML/AI platforms
  • LLM experience: recent hands-on work with at least one of: fine-tuning, prompt engineering, LLM evaluation, or model serving
  • Technical stack: strong Python across production services and data pipelines, data engineering fundamentals (Spark, Delta tables, Parquet)
  • Platform and infrastructure: Databricks (MLflow, Unity Catalog, Model Serving), AWS (EKS/Kubernetes, IAM), Terraform, GitHub Actions
  • Cross-domain flexibility: comfort working across ML, data engineering, and infrastructure. You don’t need to be expert in all three, but you contribute wherever the team needs it

Nice to have:

  • LLM evaluation frameworks (judges, graders, calibration methodology) or fine-tuning techniques (LoRA, RLHF/DPO, model distillation)
  • ML data engineering: synthetic data generation, evaluation dataset design, annotation pipelines
  • Healthcare, regulated industry, or safety-critical AI systems experience
  • Prompt optimisation frameworks (DSPy or similar), feature stores (Tecton)

#LI-KP1 #LI-Hybrid 

Annual Salary Range (ranges may vary based on skills and experience)
£120,000£150,000 GBP

How we work

We’re a mission-led, product-driven team. We move fast, stay focused and take ownership – from brief to build to impact. Debate is encouraged. Decisions are shared. We care about craft, ship with purpose, and always raise the bar.

You’ll be working with people who take their work seriously, not themselves. It takes commitment, resilience, and the drive to keep going when things get tough. Because better health outcomes are worth it. 

What you'll get

We support impact with meaningful reward. Here’s what that looks like:

  • Competitive salary and annual reviews
  • Opportunity to participate in Flo’s performance incentive scheme
  • Paid holiday, sick leave, and female health leave
  • Enhanced parental leave and pay for maternity, paternity, same-sex and adoptive parents
  • Accelerated professional growth through world-changing work and learning support
  • In-person collaboration and work in a hybrid model, with 3 days per week spent in the office
  • 5-week fully paid sabbatical at 5-year Floversary
  • Flo Premium for friends & family, plus more health, pension and wellbeing perks

Diversity, equity and inclusion

Our strength is in our differences. At Flo, hiring is based on merit, skill and what you bring to the role – nothing else. We’re proud to be an equal opportunity employer, and we welcome applicants from all backgrounds, communities and identities. Read our privacy notice for job applicants.

Top Skills

Ansible
Spark
AWS
Azure
Databricks
Docker
GCP
Kubernetes
Mlflow
Python
Terraform
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco, CA
492 Employees
Year Founded: 2015

What We Do

Flo is the essential female health partner. We are building a better future for female health by helping women harness the power of their body signals. --------------- Flo in numbers: --------------- #1 period tracker in the USA by active audience #1 most downloaded female health app worldwide 47 million monthly active users 240 million installs 28M women got pregnant while using Flo (the number indicates users who logged pregnancy in app during 2015-2022) Flo prioritizes safety and keeps a sharp focus on being the most trusted digital source for women's health information. Flo intensively uses Data Science and AI to deliver the most personalized content and services available. Flo is currently available in 22 languages on iOS and Android. 55% of users are located in the US and Europe. We are committed to putting privacy of our users first. Our users trust our technology to keep their sensitive information private and we are responsible for providing a safe and secure platform for them to use. We regularly carry out privacy assessments to monitor risks including data protection impact assessments and third-party risk assessments. In March 2022 Flo completed an external, independent privacy audit which confirmed there are no gaps or weaknesses in our privacy practices.

Similar Jobs

BNY Logo BNY

Machine Learning Engineer

Fintech • Financial Services
In-Office
2 Locations
41739 Employees
In-Office
Henley-on-Thames, Oxfordshire, England, GBR
4700 Employees
In-Office
2 Locations
2359 Employees
120K-130K Annually
Easy Apply
In-Office
9 Locations
2209 Employees

Similar Companies Hiring

Peaksware Thumbnail
Fitness • Music • Software
Louisville, CO
245 Employees
WHOOP Thumbnail
Wearables • Sports • Healthtech • Hardware • Fitness
Boston, MA
500 Employees
SRAM, LLC Thumbnail
Transportation • Sports • Software • Mobile • Hardware • Fitness • Esports
Chicago, IL
3800 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account