Senior AI Platform Engineer

The role

You will evolve and maintain the cloud-native AI platform used by engineers and data scientists to build and deploy ML and AI products. You will design internal tools, APIs and MLOps practices, and work with Data Science and Infrastructure teams.

What you will do

• Evolve and maintain the Kubeflow, Feast and Spark-on-Kubernetes infrastructure
• Design internal tools, APIs and abstractions so distributed teams can own their ML lifecycle
• Adapt software engineering best practices (CI/CD, versioning, testing) to ML workflows
• Drive MLOps practices and define lifecycle requirements for LLMOps
• Partner with Infrastructure and Data squads to integrate ML artifacts into the Data Catalog, Privacy and Governance frameworks
• Inspire colleagues to care for their own wellbeing

Required qualifications

• Deep experience with CI/CD, Infrastructure as Code (Terraform/Crossplane) and Observability in ML/AI
• Hands-on mastery of Kubeflow, Spark, Kubernetes and the AWS ecosystem
• Strong Python skills for building reusable libraries, APIs and tooling
• Ability to partner with Data Science teams and translate complex use cases into scalable solutions
• Platform-as-a-product mindset with focus on developer experience
• Systems thinking across ML infrastructure, data cataloging and governance

Benefits

• Free Wellhub Gold+ membership with onsite gyms, digital fitness, and wellness resources; up to three family members can be added
• Wellz emotional wellbeing program with 52 individual therapy sessions per year and on-demand content
• Health, dental and life insurance
• Hybrid and remote options; home office reimbursement
• Flexible schedule defined with leaders
• Paid vacation after 6 months, 3 extra days off per year, plus 1 additional day per year of tenure (up to 5), and a birthday holiday
• 100% paid parental leave for all new parents, with extended leave and ramp-back period for parents giving birth
• Career growth platforms, development roadmap and internal opportunities

About the company

Wellhub is a workplace wellness company whose platform connects employees to partners for fitness, mindfulness, therapy, nutrition and sleep through one subscription. It is headquartered in NYC with team members in Europe, North America and South America.