Shield AI
From Shield AI's careers page
Status: listed on the employer's own careers page when we last checked (daily). Confirm on their site before applying.
We are looking for a Senior Software Engineer to build the backend services that make those workflows reliable.
Shield AI is a venture-backed defense-tech company with the mission of protecting service members and civilians with intelligent systems. Its products include Hivemind autonomy software, V-BAT and X-BAT aircraft, and Aechelon simulation and synthetic reality technologies. With offices and facilities across the U.S., Europe, the Middle East, and Asia-Pacific, Shield AI’s technology actively supports operations worldwide. For more information, visit www.shield.ai. Follow Shield AI on LinkedIn, X, Instagram, and YouTube.
Model development produces experiments, checkpoints, evaluation results and many versions of the same model. Keeping those records connected lets engineers reproduce results, compare approaches and move a trained model into use. We are looking for a Senior Software Engineer to build the backend services that make those workflows reliable. You will connect training runs to their data, code and configuration, manage model artifacts and help machine-learning engineers spend less time reconstructing what happened in an experiment.
Build experiment-tracking and model-registry services, integrating tools such as MLflow where appropriate.
Connect exact dataset, code and configuration versions to training runs, checkpoints and model versions.
Develop APIs and adapters that connect training jobs, data services and artifact storage.
Make interrupted or failed workflows recoverable through reliable state handling, retries and useful diagnostics.
Make checkpoints, model artifacts and evaluation results easy to find and retrieve for further training, evaluation or deployment.
Work directly with machine-learning engineers to simplify their workflows and reduce manual record keeping.
Checkpoints and evaluation evidence remain available when jobs fail or workflows change.
Developers can compare experiments and retrieve model artifacts through consistent APIs.
Strong software engineering skills in Python or a comparable backend language.
Experience designing APIs, metadata models and persistent state for services used by other engineers.
Experience handling storage, partial failures, retries and consistency in distributed systems.
Understanding of training runs, checkpoints and model versioning.
Experience integrating Kubernetes jobs, object storage or machine-learning workflow systems.
Experience with reproducibility, data provenance or evaluation records.
As a Senior Staff Engineer, you will help build the reference architecture used internally by our engineering teams and delivered to customers building their own AI capabilities. Your work will directly impact how quickly researchers can iterate, how efficiently compute resources are utilized, and how AI systems move from experimentation to operational deployment.
This is an opportunity to work at the intersection of modern AI, distributed systems, Kubernetes, and defense, building a platform that enables the next generation of autonomy systems while leveraging the best of the open-source AI ecosystem.
A free account lets you apply, save this job, see how well you match and check your resume against it.
Scores your resume against this job the way resume scanners do: the job's hard and soft skills, searchability, recruiter tips and formatting.
Create a free account to see your score. Your resume stays in this browser until then, and we never store it.
Already a member? Sign in
