Staff Software Engineer ML Platform

Skills

About the Role

You will build backend services that connect machine-learning training runs with their datasets, code, configuration, checkpoints, and model versions. You will develop APIs and storage integrations, make workflows recoverable, deliver observable tested services, and simplify experiment-management workflows for machine-learning engineers.

Requirements

• Experience building production backend services for machine-learning platforms or data-intensive workflows
• Strong software-engineering skills in Python or a comparable backend language
• Experience designing APIs, metadata models, and persistent state for engineer-facing services
• Experience handling storage, partial failures, retries, and consistency in distributed systems
• Understanding of training runs, checkpoints, and model versioning
• Ability to work directly with machine-learning users and deliver maintainable workflow software

Responsibilities

• Build experiment-tracking and model-registry services
• Connect dataset, code, and configuration versions to training runs, checkpoints, and model versions
• Develop APIs and adapters for training jobs, data services, and artifact storage
• Make failed and interrupted workflows recoverable through state handling, retries, and diagnostics
• Enable retrieval of checkpoints, model artifacts, and evaluation results
• Work with machine-learning engineers to simplify workflows and reduce manual record keeping
• Deliver tested, observable services with compatible upgrades across cloud and customer-managed environments

Benefits

• Bonus
• Benefits
• Equity