Flo is the world’s #1 health app on a mission to build a better future for female health. Backed by a $200M investment led by General Atlantic, it became the first product of its kind to reach a $1B valuation in 2024. It has 6M paid subscribers and the highest-rated experience in the App Store’s health category, focusing on AI-powered, privacy-first, clinically backed digital health solutions.

The job involves working with a world-class team of product builders, engineers, data scientists, user researchers, designers, and medical experts to build an app that provides accurate cycle predictions and personalized, medically credible health tips using machine learning and AI.

The role entails owning the end-to-end AI/ML pipeline, including CI/CD processes for ML and AI engineering workflows. Responsibilities include managing code, versioning datasets, models, AI agents, and production endpoints to enable effective collaboration, experimentation, and scaling across traditional ML and modern AI applications.

The AI Platform team acts as a central enabler of machine learning and AI across the organization, reducing operational overhead and maximizing ROI from ML use cases. The team builds and maintains infrastructure like feature stores, model deployment pipelines, experiment tracking, and monitoring frameworks, ensuring scalable, high-quality solutions that accelerate time-to-market while ensuring compliance and high performance.

Your Experience

• 4+ years of industry experience building and deploying production-grade AI/ML systems.
• Recent hands-on engineering experience with AI/LLM infrastructure and tooling.
• Strong proficiency with Python, and familiarity with Scala, Go, or Rust.
• Cloud platform expertise with AWS, GCP, or Azure, including AI-specific services like SageMaker, Vertex AI, Azure AI.
• Experience with Databricks platform, including Unity Catalog, MLflow, and Databricks Machine Learning.
• Modern ML infrastructure experience with tools like MLflow or similar.
• Containerization and orchestration experience with Docker, Kubernetes, and ML-specific operators.
• AI model serving experience with modern inference servers and API gateways for AI applications.

Nice to have

• Infrastructure as Code experience with Terraform, Ansible, or other tools.
• Distributed computing experience with Databricks, Ray, or Spark for large-scale AI workloads.
• AI safety and governance experience with model evaluation, bias detection, and responsible AI practices.
• Multi-modal AI experience with vision-language models, speech processing, or other modalities.
• Experience at a Tier-1 product company or working in a product organization.

Responsibilities

• Design and maintain automated pipelines for traditional ML and modern AI systems including LLMs, multimodal models, and AI agents.
• Implement CI/CD practices for model training, fine-tuning, evaluation, and deployment across diverse AI workloads.
• Orchestrate deployment of models, AI agents, and inference endpoints with automated testing and rollback capabilities.
• Build and maintain infrastructure for large language model inference, fine-tuning, and serving at scale.
• Implement monitoring for model performance, drift detection, AI safety metrics, and responsible AI compliance.
• Evolve operational processes to support traditional ML and advanced AI technologies.

Work environment and benefits

• 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 and learning support.
• Flexible office and home working with up to 2 months a year working abroad.
• 5-week fully paid sabbatical at 5-year work anniversary.
• Flo Premium for friends and family, plus other health, pension, and wellbeing perks.

Diversity, equity and inclusion