Senior AI/ML engineer
Core
Building the GenAI platform that powers every AI feature at Flo, including production-grade medical safety judges, fine-tuning pipelines, and real-time personalization.
Role type
Senior IC machine-learning engineer (LLM infrastructure)
Builds
GenAI platform, LLM-based evaluation judges, proprietary fine-tuned health models
Domain
Digital health, AI/ML infrastructure
Deliverable
production ML models
Required skills
Python, data engineering (Spark, Delta tables, Parquet), LLM fine-tuning, model serving, prompt engineering, LLM evaluation, Terraform, AWS (EKS, S3, IAM), Databricks (MLflow, Unity Catalog), GitHub Actions
Preferred skills
LLM evaluation frameworks (judges, graders, calibration), fine-tuning techniques (LoRA, RLHF/DPO, model distillation), synthetic data generation, prompt optimization (DSPy), feature stores (Tecton), healthcare/safety-critical AI experience
Technologies
Python, Spark, Delta tables, Parquet, MLflow, Llama, Gemma, MedGemma, Databricks, AWS, Terraform, GitHub Actions, Unity Catalog, EKS, S3, IAM
Responsibilities
Build and scale Judge-as-a-Service and evaluation orchestration; develop LoRA/SFT/preference optimization pipelines; build synthetic Q&A generation and evaluation data pipelines; maintain Terraform-managed AWS infrastructure and CI/CD pipelines; collaborate with cross-functional teams to develop internal SDKs and APIs
Seniority
Senior, hands-on IC