Staff Engineer, AI Platform & Architecture
$190k - $290kShield AI
Role Description
The Staff Engineer, AI Engineering is a senior individual contributor responsible for translating the enterprise AI engineering roadmap into scalable platform architecture, reusable technical patterns, and production-grade shared services. Reporting to the Director, AI Engineering, this role provides deep technical leadership across AI enablement, responsible AI controls, observability, cost attribution, and reusable component strategy. The Staff Engineer acts as the connective technical tissue across Engineering, IT, Security, Legal, Data, and business unit teams - setting standards, creating reference implementations, and guiding teams toward consistent, secure, measurable AI adoption without relying on direct authority. Success is defined by high-quality platform components adopted across teams, clear architecture and governance patterns, measurable productivity and cost outcomes, and effective mentorship of engineers building AI-enabled capabilities.
What you'll do:
- AI Platform Architecture & Standards
- Define and evolve enterprise AI architecture patterns for LLM integration, retrieval-augmented generation, agentic workflows, prompt orchestration, and workflow automation.
- Create reference architectures, design reviews, decision records, and implementation guidance that enable consistent AI development across business units.
- Serve as a technical authority for AI platform decisions, including model selection, integration approaches, data boundary enforcement, and lifecycle management.
- Evaluate emerging AI technologies and recommend fit-for-purpose adoption paths aligned to security, operational, and enterprise architecture requirements.
- Partner with product, platform, and business technology teams to identify common needs and convert them into reusable engineering patterns.
- Reusable Components & Shared Services
- Design and build reusable AI components such as connectors, agents, skill templates, prompt libraries, data pipelines, integration adapters, and service APIs.
- Lead technical design for shared platform services for AI observability, logging, usage metering, evaluation, and lifecycle management.
- Establish quality, versioning, deprecation, documentation, and contribution standards for the shared AI component catalog.
- Guide teams through adoption of shared components, balancing standardization with practical implementation needs.
- Identify opportunities to eliminate duplicate AI engineering efforts through consolidation, abstractions, and platformization.
- Responsible AI Engineering & Governance
- Architect engineering controls for access management, data classification enforcement, prompt safety, output validation, audit logging, and policy adherence.
- Partner with Security, Legal, and compliance stakeholders to embed responsible AI requirements into development and deployment pipelines.
- Design model and agent lifecycle governance patterns, including version tracking, evaluation, drift monitoring, rollback, and deprecation workflows.
- Build technical dashboards and telemetry that expose adoption, risk, performance, and governance compliance across AI-enabled systems.
- Represent engineering considerations in AI governance reviews and translate policy requirements into implementable technical standards.
- Productivity, Measurement & Technical Leadership
- Develop AI-assisted workflow patterns that improve individual productivity, team collaboration, knowledge retrieval, meeting intelligence, document generation, and task automation.
- Design measurement approaches that connect AI usage to time savings, quality improvement, error reduction, capacity creation, and business value.
- Partner with Finance and platform teams to develop cost metering, showback/chargeback, and optimization mechanisms for AI services.
- Mentor senior and mid-level engineers, raise engineering quality, and lead complex cross-functional technical initiatives from concept through production.
- Contribute to communities of practice, internal enablement material, and technical evangelism for enterprise AI engineering standards.
Qualifications
- Progressive experience in enterprise software engineering, AI platform engineering, data platform engineering, or digital workplace technology roles.
- Deep hands-on understanding of generative AI, large language model integration, RAG architectures, agentic AI patterns, prompt orchestration, and production AI system design.
- Experience designing shared platform services, reusable component libraries, APIs, integration frameworks, or developer enablement platforms used by multiple teams.
- Strong architecture judgment across security, reliability, scalability, observability, maintainability, and operational cost tradeoffs.
- Experience implementing or contributing to AI governance controls such as access management, data classification, audit logging, model lifecycle management, and compliance-aware development practices.
- Ability to influence technical direction across matrixed teams through architecture reviews, written guidance, reference implementations, and hands-on collaboration.
- Experience defining metrics, telemetry, or attribution mechanisms for adoption, productivity, cost, quality, or operational performance.
- Strong written and verbal communication skills with the ability to explain complex AI engineering concepts to technical and non-technical audiences.
Preferred Qualifications
- Experience in regulated, defense-adjacent, security-sensitive, or data-governed environments.
- Background in MLOps, AI observability, model evaluation frameworks, agent evaluation, and production monitoring.
- Familiarity with enterprise AI tooling ecosystems including copilot platforms, workflow automation suites, vector databases, and enterprise search/RAG platforms.
- Hands-on experience with enterprise data platforms such as Databricks, Snowflake, lakehouse architectures, or comparable data foundations.
- Experience implementing usage metering, cost allocation, showback/chargeback, or AI spend optimization capabilities.
- Track record of mentoring engineers and raising technical standards without relying on direct management authority.
- Advanced degree in Computer Science, Engineering, Data Science, or a related technical field.
Compensation
$190,000 - $290,000 a year
Employee Offer Package
- 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)
Equal Opportunity Statement
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.
$253.3k - $354.6k
...standard retrieval and ranking architectures, leveraging modern deep... ..., user engagement, and platform growth at a massive scale.... ...ll Have Impact As a Staff Machine Learning Engineer , you will own the technical... ...to influence key AI-driven systems across Reddit...SuggestedRemote jobFull timeFor contractorsWork experience placementFlexible hours- ...Designing and owning a new internal AI platform, the full-time Staff Engineer, AI Platform will create foundational architecture and tooling to enable teams across the company to develop AI-powered workflows, all while working remotely. Key responsibilities Design and...SuggestedFull timeRemote work
$163k - $253k
...the growing demands of future AI/ML workloads. Our team is... ...designing and developing scalable platforms that can effectively handle... ...with both hardware and software engineers to identify and address the... ...with remote or cloud IDE architecture (Gitpod, Coder, Codespaces, or...SuggestedWork at officeRemote workFlexible hours$100k - $140k
...Department Department: BSD CTD - Platform Engineering - GDC The Center for... ...implementation, security automation, & AI/ML infrastructure management... ...incidents. For monitoring, staff wrangle disparate system... ...are for hybrid cloud architecture on-premises and in commercial...SuggestedWork experience placement- Summary: The Staff Engineer, Composer Platform will provide technical leadership for Finalsite’s core CMS... ...institutions nationwide. This role will lead architectural modernization efforts, improve... ...best practices Utilize AI-assisted development tools to support...SuggestedPermanent employmentFull timeRemote work
- ...Company Description Who we are Moveworks is the Agentic AI Assistant platform that empowers the entire workforce. Our platform enables... ...’s leading workflow automation with Moveworks’ Reasoning Engine and natural language capabilities, we deliver the AI platform...Full timeWork at officeRemote workFlexible hours
$170k - $240k
...We are seeking a high-caliber Full-Stack Engineer to lead the development of a web-based simulation platform. Our goal is to build a high-performance "Engineering... ...Gatik’s well-defined, fixed route ODDs and hybrid architecture, we are able to hyper-optimize our models with...Odd jobFull timeWork at office$230k - $300k
...the Content Intelligence platform shaping the future content... ...information powered by advanced AI, recommendation systems,... ...Role We are seeking a Staff Machine Learning Engineer to provide technical... ...Technical Leadership & Architecture Lead the design of next...Full timeLocal areaWork from home- ...About ShiftKey ShiftKey is a platform that is disrupting the way... ...are building toward being an AI-native company. We are past... ...why this role exists. The Staff AI Engineer owns ShiftKey’s AI... ...retrieval foundations and own architecture of everything that comes next...Full timeWork experience placementWork at officeRemote workFlexible hoursShift work
$175k - $213.5k
...and web-based learning platform that delivers... ...their careers. The engineering team collaborates closely... ...coding standards, and architectural patterns—both individually... ...Role Overview The Staff Full-Stack Engineer on... ...and interaction with AI/ML systems. Key Responsibilities...Full timeLocal areaRemote workFlexible hours- ...robots, industry-leading sensors, and an AI-powered data platform to provide customers with a unique... ...Role at a Glance Gecko Robotics’ Engineering teams build high-impact, customer-... ..., frontend applications, and system architecture, while collaborating with product...Full timeInternshipWork at officeLocal areaWork from homeFlexible hoursShift work
- ...The Client Data & Networking Platform team owns the data layer of Airbnb... ...the team shapes how frontend engineers across the company build and... ...of software development from architecture/design through implementation... ...Experience building agentic AI solutions is a plus....Remote jobFull timeCasual workLive inWork at office
$170k - $210k
...Software Engineer, Delivery Network Platform At Zipline, we are building the operating system for an autonomous... ...We also expect engineers to use AI tooling effectively as a force... ...durable abstractions, and improve the architecture around you care deeply about reliability...Full timeLocal areaImmediate startShift work- ...products. As a Senior Lead Software Engineer - ML Engineer for Agent Platform at JPMorgan Chase within the... ...for Payments Technology. You set the architecture for how agents execute, communicate... ...adoption of enterprise-authorized AI-assisted engineering practices to improve...
- ...our team at Workiva as a Staff Machine Learning Engineer! As a pivotal member of our... ...team, you'll spearhead the architecture and delivery of... ...learning solutions across our platform. Your expertise will be instrumental... ...cutting-edge Generative AI into our products. In this...Work at officeRemote work
- ...Job Description Founded in 2015, Shield AI is a venture-backed defense-tech company... ...develop and integrate the core software and platform infrastructure that powers mission-... ...5+ years experience with strong software engineering experience in robotics, autonomy, embedded...Full timeTemporary workPart timeRemote workWorldwide
- ...Who we are Moveworks is the Agentic AI Assistant platform that empowers the entire workforce.... ...automation with Moveworks’ Reasoning Engine and natural language capabilities, we... ...Implement frontier AI algorithms and architectures and help productionize them at scale...Full timeWork at officeImmediate startRemote workFlexible hours
$188k - $275k
...The Essential Cloud for AI™. Built for pioneers... ...CoreWeave delivers a platform of technology, tools,... ...platform that gives network engineers, fleet engineers, and... ...the patterns and architecture that the team builds on... ...production systems. For Staff level: demonstrated...Permanent employmentFull timeTemporary workCasual workWork at officeImmediate startRemote workFlexible hours$191k - $315k
...Overview: The Notification AI team builds large-... ...members get from the platform, driving over 50% of... ...systems, and scalable architectures that deepen LinkedIn’s... ...closely with Product, Engineering, and Data Science to deliver... ...technical leadership (Staff+) experience, including...Full timeFor contractorsWork at officeFlexible hours- ...data, applications, processes, and AI into a single, governed platform. A leader in Enterprise MCP and trusted... ...500, Workato’s cloud-native architecture connects every application, data source... ...for an exceptional Search/AI Engineer with experience in Search Relevance...Full timeRemote workFlexible hours
- ...Job Overview As a Staff AI Engineer, you define and drive the architecture of AI and agentic systems across multiple teams and product domains. This is a senior... ...AI and agentic systems, and contribute to the AI platform roadmap across teams Influence architecture decisions...Work at officeRemote workFlexible hours
$207k - $275k
...CoreWeave is The Essential Cloud for AI™. Built for pioneers by pioneers, CoreWeave delivers a platform of technology, tools, and... ...What You'll Do The Endpoint Engineering team at CoreWeave manages end... ...the end-to-end VDI platform architecture—covering cloud infrastructure...Permanent employmentTemporary workCasual workWork at officeRemote workFlexible hours$221k - $260k
...Staff Machine Learning Engineer – AI Tech Lead Location: USA The proliferation of AI and machine... ...technologies and build scalable multi-agent architectures that reason over large-scale... ...direction for Sumo Logic’s agentic AI platform and play a key role in bringing...Full timeWorldwide$115k - $125k
...Our award-winning modern core platform and cloud-based technology... ...excel. We operate with an AI-first mindset across our business... ...: We're hiring a Release Engineer to join our growing Release... ...of the Nymbus platform architecture to inform deployment decisions...Immediate startRemote workFlexible hoursAfternoon shift$180k - $220k
Role Description The Staff Engineer - ML Operations is responsible for... ...powers Danaher's next-generation AI-driven research. You will be... .... ~Partner with the platform team to maximize value from... ...while staying hands-on with architecture and delivery. Qualifications...Full time- ...Job Description Job Description Platform Staff Engineer Great people. Greater business impact.... ...Do Responsible for the execution of architecture, development, implementation and quality... ...systems that power CAI Parsable’s AI-driven platform capabilities Establish...For contractorsLocal area
$184k - $253k
Role Description As a Senior Staff Forward Deployed Engineer, AI, you will be tasked with extending SentinelOne... ...AI coding assistants, and internal platform APIs, moving from prototype to... ...~Own delivery end to end including architecture, build, testing, deployment, and the...Full timeHome officeFlexible hoursShift work- ...fluent, and development-oriented Senior Staff Engineer, Internal Platform to join our global Platform... ...tier database caching, and large-scale architectural modernization laboratory. Partnering... ...3 primitives, champions intelligent AI automation, and coordinates seamlessly...Long term contractPermanent employmentFull timeRemote workShift work
$180k - $250k
Role Description The Staff Engineer, Communications Platform will provide technical leadership for Finalsite’s... ...observability initiatives, and long-term architectural strategy across a multi-cloud... ...operational experience ~Experience using AI-assisted development tools such as...Full timeRemote workWorldwide- Role Description As a Staff Engineer on the Platform Engineering team, you will play a pivotal role in shaping... ...through effective adoption of AI-powered development workflows and systems... ...to high-performance service architectures capable of handling massive scale with...Full time
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff Engineer, AI Platform & Architecture. Be the first to apply!
- assistant engineer Remote
- staff design engineer Remote
- assistant engineering manager Remote
- senior staff systems engineer Remote
- assistant chief engineer Remote
- staff data engineer Remote
- project engineer assistant project manager Remote
- engineering aide Remote
- senior staff engineer Remote
- staff security engineer Remote









