$233K – $350K/yr
Pay vs market · Near median
This job's pay ($250,000 a year at the midpoint) is higher than about 55% of 1,227 salary-listed Software Engineer jobs in United States in the last 12 months; their median is $232,500.
Role overview
Job description
-
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.
-
Deliver tested services with observability and compatible upgrades across supported cloud and customer-managed environments.
-
Engineers can trace a model version back to the data, code, configuration and run that produced it.
-
Checkpoints and evaluation evidence remain available when jobs fail or workflows change.
-
Developers can compare experiments and retrieve model artifacts through consistent APIs.
-
Teams can reproduce results without manually reconstructing experiment history.
-
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 services used by other engineers.
-
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 software that solves their workflow problems.
-
Experience with MLflow, Weights & Biases, model registries or comparable experiment-management tools.
-
Experience integrating Kubernetes jobs, object storage or machine-learning workflow systems.
-
Experience with reproducibility, data provenance or evaluation records.
-
Experience supporting open-source software or on-premises and air-gapped deployments.
The AI Factory is the foundation for how Shield AI develops and deploys autonomy.
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.
#LI-DM2 #LF Full-time regular employee offer package: Pay within range listed + Bonus + Benefits + Equity Temporary employee offer package: Pay within range listed above + temporary benefits package (applicable after 60 days of employment) Salary compensation is influenced by a wide array of factors including but not limited to skill set, level of experience, licenses and certifications, and specific work location. All offers are contingent on a cleared background and possible reference check. Military fellows and part-time employees are not eligible for benefits. Please speak to your talent acquisition representative for more information. ### Shield AI is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, marital status, disability, gender identity or Veteran status. If you have a disability or special need that requires accommodation, please let us know.About this listing
- Source: Shield AI's official careers site (via Lever) · last confirmed on the source 3 Nov 2026
- Posted 2 Nov 2026
Jobiba does not charge job seekers. Never pay to apply or share OTPs, card or bank details.
What they are looking for
Skills & requirements
Department · Hivemind Enterprise Division
Hiring company
Shield AI
View the company profile, current openings and hiring details.
Related opportunities
Similar active roles
Relevant, currently active roles for this market.
$160K – $290K/yr
$170K – $310K/yr
Senior/Staff Software Engineer with C++ - Drivers, Diagnostic, & Embedded Software (San Diego, CA)
Philips India$126K – $201K/yr
$200K – $300K/yr
$150K – $230K/yr
🔔 Get job alerts
New Software Engineer ML Platform jobs in San Diego, California, United States, straight to your inbox.
No spam · Unsubscribe anytime