Staff Engineer, ML

We are looking for an experienced ML/AI engineer to anchor our ML Platform team. You will lead the effort to productize our Large Language Models, transforming experimental code into robust, efficient services. Your goal is to build the infrastructure that allows our Data Scientists to move fast without breaking production.

What you'll be doing:
Model Development & Optimization

Fine-tuning & Engineering: Develop and fine-tune LLMs/classic models for specific medical tasks (condition detection, dialogue systems) using internal datasets.
Research to Production: Partner with Data Scientists to develop experimental code into scalable, production-ready modules.
Platform Engineering & Infrastructure

Pipeline Orchestration: Build and maintain complex ML workflows using Kubeflow Pipelines (KFP) or similar orchestration tools.
Internal Tooling: Develop and manage the internal Python ecosystem-libraries, SDKs, and utilities-that the Data Science team uses for daily development.
CI/CD & Automation: Write and maintain CI/CD scripts to automate the testing, versioning, and deployment of machine learning artifacts.
Production Standards & Integration

System Integration: Work with backend developers to integrate trained models/agents into the core application architecture.
Code Quality: Enforce high engineering standards through code reviews, ensuring that research code meets production reliability and maintainability requirements.
Requirements Engineering: Translate evolving data science requirements into concrete infrastructure and platform features.