Senior AI Platform Engineer
Posted on August 25, 2025 (about 12 hours ago)
Job Description
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.
With 6M paid subscribers and the highest-rated experience in the App Store’s health category, Flo is building the next generation of digital health – AI-powered, privacy-first, clinically backed – to help users know their body better.
The job
Flo's team includes product builders, engineers, data scientists, growth hackers, user researchers, designers, and medical experts building an app that generates long-term value to millions of users worldwide through accurate cycle predictions and personalized, medically credible health tips using machine learning and AI.
As a Senior AI Platform Engineer, you will own the end-to-end AI/ML pipeline, including CI/CD processes for ML and AI engineering workflows. You'll manage code, version datasets, foundation and fine-tuned models, AI agents, and production endpoints enabling ML Engineers to collaborate effectively, experiment efficiently, and scale rapidly across traditional ML workloads and modern AI applications.
The AI Platform team delivers scalable, high-quality solutions by maintaining critical infrastructure like feature stores, model deployment pipelines, experiment tracking systems, and monitoring frameworks ensuring compliance and high performance.
Your Experience
Must have:
- 4+ years of industry experience building and deploying production-grade AI/ML systems.
- Hands-on engineering experience with AI/LLM infrastructure and tooling.
- Strong proficiency with Python; familiarity with Scala, Go, or Rust.
- Cloud platform expertise with AWS, GCP, or Azure, including AI-specific services (SageMaker, Vertex AI, Azure AI).
- Databricks platform experience with Unity Catalog, MLflow, and Databricks Machine Learning for end-to-end AI/ML workflows.
- Modern ML infrastructure experience with tools like MLflow or similar platforms.
- Experience with containerization & orchestration using 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 IaC tools.
- Distributed computing experience with Databricks, Ray, or Spark for large-scale AI workloads.
- AI safety & 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 Tier-1 product companies or similar product organization experience.
Responsibilities
- Design and maintain automated pipelines for traditional ML models and modern AI systems including LLMs, multimodal models, and AI agents.
- Implement robust CI/CD practices for model training, fine-tuning, evaluation, and deployment across diverse AI workloads.
- Orchestrate seamless 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 comprehensive monitoring for model performance, drift detection, AI safety metrics, and responsible AI compliance.
- Evolve operational processes to support both traditional ML and cutting-edge AI technologies.
Culture and Benefits
Flo is a mission-led, product-driven team that moves fast, stays focused, and takes ownership from brief to build to impact. The company values debate, shared decisions, craftsmanship, purposeful shipping, and always raising the bar.
Employees work with committed, resilient colleagues focused on better health outcomes. The company offers competitive salary with annual reviews and participation in a performance incentive scheme, paid holiday, sick leave, and female health leave, enhanced parental leave and pay, accelerated professional growth, flexible working arrangements including work abroad, a fully paid sabbatical after 5 years, and various health, pension, and wellbeing perks.
Flo is proud to be an equal opportunity employer welcoming applicants from all backgrounds, communities, and identities, emphasizing merit and skills in hiring.
How to Apply
Applications can be submitted through the job board page where the job is posted. Candidates are required to fill out a form including personal details such as first name, last name, email, phone, location, and upload a resume/CV. Cover letter submission is optional but available. The application form also includes questions related to the candidate's experience with CI/CD pipelines, Databricks, Infrastructure as Code, and more. Candidates must confirm compliance with privacy policies and location requirements for the role based in Lithuania with hybrid work obligations.
The position requires willingness to travel to the office in Vilnius three days a week as part of the hybrid work model.
For further details, candidates are encouraged to review the privacy notice for job applicants on Flo's website.