Staff Engineer, Data Platform (R5659)

F7d8749a; San Diego (CA); Presencial; USD 150.000 - 230.000; Tempo integral

Benefícios oferecidos por esta vaga

Equity

Benefits

Bonus

Resumo da vaga

Shield AI in San Diego is seeking a Staff Engineer to build the services behind our AI Factory data workflows. You will design and implement data APIs, dataset versioning, search and metadata systems that scale with growing datasets and evolving schemas.

Collaborate with ML engineers to turn recurring data needs into reusable tools, and deliver reliable data services with strong diagnostics and support for large-scale datasets across diverse model workflows.

Qualificações

• Experience designing production data platforms or data-intensive backends.
• Strong Python or equivalent backend language.
• Experience with data indexing, versioning, metadata, and access control.
• Ability to lead technical design and implement substantial parts of a data system.
• Ability to translate ML data needs into reusable data services.

Responsabilidades

• Build dataset discovery, catalog and retrieval services, with LanceDB integration.
• Create APIs to pin dataset versions and control access.
• Track data provenance and changes to reproduce results.
• Design schema validation and migration for historical datasets.
• Define identifiers and metadata relationships linking datasets to experiments.
• Collaborate with ML engineers to turn needs into reusable tools.
• Deliver reliable data services with upgrades and diagnostics.

Conhecimentos

Data platform design

Python backend

API design

Data indexing and versioning

Schema evolution

Leadership & collaboration

Ferramentas

LanceDB

Arrow/Parquet

Kubernetes

SQL / metadata storage

Descrição da vaga

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 https://shield.ai/.

JOB DESCRIPTION:

Reproducing a model’s results starts with knowing exactly which data it used. Shield AI’s AI Factory gives machine-learning engineers tools to find, access and reuse training data across internal programs and customer deployments. We are looking for a Staff Engineer to build the services behind those workflows. You will design and implement data APIs, dataset versioning, search and metadata systems that remain useful as datasets grow and schemas change. This role combines technical leadership with substantial hands‑on software development.

What you'll do:

• Build dataset discovery, catalog and retrieval services, including LanceDB integration for indexing and search.
• Create APIs that let developers pin an exact dataset version and retrieve only the data they are authorized to access.
• Track where data came from and how it changed so engineers can understand and reproduce training results.
• Design schema validation, compatibility and migration mechanisms that preserve access to historical datasets.
• Define common identifiers and metadata relationships that connect datasets to experiments and model artifacts.
• Work with machine-learning engineers to understand how they create and use datasets, then turn recurring needs into reusable tools.
• Deliver reliable data services with tested upgrades, clear diagnostics and support for large datasets.

Key outcomes:

• Developers can find and retrieve the right dataset without assembling a new data pipeline for each experiment.
• Training runs can reference exact dataset versions and trace them back to their sources.
• Schema changes have explicit compatibility or migration paths for existing users.
• Data services work across multiple model-development workflows and deployment environments.

Required qualifications:

• Experience designing and implementing production data platforms or data-intensive backend services.
• Strong software engineering skills in Python or a comparable backend language.
• Experience with data indexing, versioning, metadata, storage and access control.
• Experience evolving schemas or APIs while supporting existing users and historical data.
• Ability to lead technical design and implement substantial parts of the system.
• Ability to work with data or machine-learning users and translate their needs into reliable services.

Bonus qualifications and relevant technologies:

• Experience in any of the following is beneficial but not required:
  • Experience with LanceDB, Arrow, Parquet or comparable columnar and vector-data systems.
  • Experience with dataset catalogs, provenance or reproducible machine-learning workflows.
  • Experience applying domain models or ontologies to practical data relationships and schema evolution.
  • Experience deploying data services on Kubernetes or in on-premises and disconnected environments.

Why join us:

The Data Platform is foundational to how Shield AI builds, evaluates, certifies, and improves autonomy. This role offers the opportunity to shape both a core internal platform and a reference architecture delivered into demanding customer environments. Your work will determine how effectively engineers and agents can find trusted context, understand provenance, aggregate data from distributed systems, and turn operational experience into better intelligent systems. You will work at the intersection of data systems, AI infrastructure, autonomy, distributed computing, and defense while helping establish the architecture that supports the next generation of mission‑critical AI development.

Compensation:

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.

Equality:

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.