Find your next role
Discover roles across ambitious teams
17,767 jobs
- More jobs
Shield AI
Staff Software Engineer ML Platform
San Diego, California · On-site · Full time · Senior level
$200k–$300k / year
About the role
Build backend services for experiment-tracking and model-registry to support ML workflows. Connect training runs to data, code, and configurations for reproducibility.
What you’ll do
• Build experiment-tracking and model-registry services. • Connect datasets, code, and configs to training runs. • Develop APIs for training jobs and artifact storage. • Ensure recoverable workflows with retries and diagnostics. • Work with ML engineers to simplify workflows.
What you’ll bring
• Experience building backend ML platform services. • Strong Python or comparable backend skills. • Design APIs and persistent state for engineering services. • Handle storage, failures, retries in distributed systems. • Work authorization for employment.
Life at Shield AI
Offers bonus, benefits, equity, and supports equal opportunity employment.
About Shield AI
Shield AI is a defense-tech company developing intelligent systems to protect service members and civilians worldwide.