Staff Software Engineer ML Platform (6244)
Core
Build backend services for experiment tracking and model registry to connect training runs with data, code, and configurations.
Role type
Senior Software Engineer (ML Platform)
Builds
Reliable experiment-tracking and model-registry services
Domain
Defense technology / Machine Learning Infrastructure
Deliverable
production ML models
Required skills
Backend development, API design, Distributed systems, Data versioning, Cloud infrastructure, Observability, State management, Integration patterns, ML workflow orchestration
Responsibilities
Build experiment-tracking and model-registry services integrating tools like MLflow; Connect dataset, code, and configuration versions to training runs and model artifacts; Develop APIs and adapters connecting training jobs, data services, and artifact storage; Implement reliable state handling and retries for interrupted or failed workflows; Ensure easy retrieval of checkpoints and evaluation results; Deliver tested services with observability across cloud and customer-managed environments.
