Staff AI Platform Engineer

Code Metal
Boston Hub +2 more
Hybrid

Who this role is best for

A natural match if you have experience leading AI platform infrastructure and deploying production agentic systems.

Best fit for

  • Candidates with experience leading AI platform infrastructure and deploying agentic systems in production
    — “Staff-level technical leadership: owned architecture across multiple systems or teams”
  • Individuals with expertise in GPU inference serving and model gateways for LLMs
    — “Deploy, benchmark, and tune production inference for open-weight models on vLLM, SGLang, and TensorRT-LLM”
  • Candidates who have built context-engineering services for agents using retrieval-augmented generation
    — “Experience with context engineering: retrieval-augmented generation, embeddings, vector or hybrid search, and memory for agents”

Things to consider

  • This role requires a balance of hands-on engineering and strategic leadership responsibilities
    — “Set the technical direction and architecture for Code Metal's AI platform and lead the team building it”
  • Candidates must be prepared to work with both engineers and researchers across multiple teams
    — “You'll work closely with our Applied AI Research team, and you'll sometimes run experiments yourself”

How to stand out

  • Highlight your work on model gateways with routing, auth, and failover in your resume and interviews
    — “Own the model gateway that teams use to reach self-hosted and commercial models”
  • Emphasize your experience with OpenTelemetry tracing and AI service SLOs in your application materials
    — “Experience instrumenting services (for example, with OpenTelemetry tracing and metrics) and operating AI services against SLOs”
  • Showcase your ability to design for productization with multi-tenancy and versioned APIs
    — “Design for productization from day one (multi-tenancy, versioned APIs, security, and deployment in customer and air-gapped environments)”
Pace · Fast PacedCollaboration · HighAutonomy · MediumDecision Impact · Team

Derived from job-description analysis by Serendipath's career intelligence engine.

What success looks like

  • architected and built AI enablement stack
  • deployed and tuned production inference
  • designed reusable agent harnesses
Typical background
AI platform engineering experience

Skills & requirements

Required

AI Platform EngineeringProduction-grade PythonDistributed SystemsFormal MethodsModel Serving And Inference

Preferred

Cross-functional CollaborationProductization Of AI Platforms

Stack & domain

PythonPlatform EngineeringAPI DesignDistributed SystemsContainersKubernetesCI/CDTestingAI ResearchFormal MethodsOpentelemetryAI ExperimentationLeadershipCommunicationTeamworkAISoftware Engineering

About the role

Original posting from Code Metal via Ashby

ABOUT CODE METAL

Code Metal is the leader in automated software engineering you can trust. As AI writes more of the world's code, the bottleneck in software has shifted from writing code to verifying it works, and AI cannot verify its own work with certainty. Code Metal takes a fundamentally different approach: constrain AI to what it does reliably, verify every step independently of the model using formal methods, and keep engineers in the loop on the decisions that matter. The result isn't code that probably works — it's code that is provably correct, with auditable proof. Customers including the U.S. Air Force, L3Harris, RTX, and Toshiba use Code Metal to modernize legacy code, optimize performance on real hardware, and move prototypes to production, fast. Founded in 2023 with offices in Boston and San Francisco, Code Metal is funded by Accel, Salesforce Ventures, B Capital, Smith Point Capital, J2 Ventures, Shield Capital, Overmatch, RTX, and others.

Learn more at codemetal.ai http://codemetal.ai.

The Role

Code Metal's engineering teams are building AI-driven code transpilation and AI-enabled mission planning and wargaming. Both need the same foundations: models to serve, agents to run, context to manage, and results to measure. Our AI Platform team builds those foundations.

As a Staff AI Platform Engineer, you'll be the technical lead of this new four-person team. You'll architect and build the AI enablement stack our engineers depend on, from GPU inference serving and a model gateway up through agent harnesses, context engineering, observability, and AI experimentation management. It starts as an internal platform, but we're building it to product standard.

This is an engineering role first. Most of your time goes to designing, building, and operating production systems. You'll also need solid data science and AI research fundamentals: you'll work closely with our Applied AI Research team, and you'll sometimes run experiments yourself when a platform decision needs evidence.

Core Responsibilities

  • Set the technical direction and architecture for Code Metal's AI platform and lead the team building it. Own the design docs and RFCs, help with build-vs-buy decisions, and mentor the team.
  • Deploy, benchmark, and tune production inference for open-weight models on vLLM, SGLang, and TensorRT-LLM.
  • Own the model gateway that teams use to reach self-hosted and commercial models, with consistent auth, routing, failover, quotas, and cost attribution.
  • Design reusable agent harnesses and orchestration primitives that product teams can compose into reliable, verifiable workflows instead of rebuilding them for each product.
  • Build context-engineering services for memory, retrieval, and data discovery, so agents get the right information within their context and cost budgets.
  • Instrument the stack end to end with OpenTelemetry traces and service metrics, and build the experiment-tracking and artifact layer that lets engineers and researchers reproduce and compare results.
  • Design for productization from day one (multi-tenancy, versioned APIs, security, and deployment in customer and air-gapped environments), and partner with Applied AI Research, product teams, and DevOps so the platform stays aligned with what they need.

Required Qualifications

  • Production-grade Python and strong platform engineering fundamentals: API and service design, distributed systems, containers and Kubernetes, CI/CD, and testing.
  • Shipped production agentic systems, with a clear sense of where they break and how to make them reliable.
  • Experience with context engineering: retrieval-augmented generation, embeddings, vector or hybrid search, and memory for agents, ideally over code or large technical corpora.
  • Experience instrumenting services (for example, with OpenTelemetry tracing and metrics) and operating AI services against SLOs.
  • Solid data science and AI research fundamentals: how transformers and LLM inference work, experiment design, benchmarking, and model evaluation. Working familiarity with PyTorch and Hugging Face, and experience fine-tuning, evaluating, or serving language models.
  • Staff-level technical leadership: owned architecture across multiple systems or teams, written design docs and RFCs, turned ambiguous needs from several internal customers into a roadmap, and mentored engineers.

Preferred Qualifications

  • Highly desirable: Production experience running an LLM gateway or proxy such as SMG or Bifrost, or equivalent experience building an API gateway, including routing, auth, rate limiting, quotas, failover, and cost attribution.
  • Hands-on experience deploying and tuning LLM inference engines such as vLLM, SGLang, or TensorRT-LLM on GPU infrastructure, with measurable gains in throughput, latency, or cost per token.
  • Familiarity with inference optimization: speculative decoding, prefix caching, tensor/pipeline/expert parallelism, disaggregated prefill and decode, GPU profiling.
  • Experience building evaluation harnesses for LLMs and agents, and experiment-tracking or artifact systems such as MLflow or Weights & Biases.
  • Experience taking an internal platform to an external product: multi-tenancy, SDKs, versioned APIs, and documentation.
  • Experience deploying AI systems on-prem or in air-gapped or classified environments, or in regulated domains such as defense or aerospace.

Experience Level

Typically 8+ years of software engineering experience, including 4+ years building and operating ML or LLM systems in production, with demonstrated Staff-level scope and impact. Equivalent depth of experience is valued over a rigid year count.

BENEFITS

  • Pay depends on experience, but we strive to be at the upper end of the salary range
  • Health care plan with 100% premium coverage, including medical, dental, and vision
  • 401k with 5% matching
  • Paid Time Off (uncapped vacation, plus sick and public holidays)
  • Flexible hybrid or remote work arrangement
  • Relocation assistance for qualifying employees

Wage Transparency - The salary range for this role is not a guarantee of compensation or salary, as the final offer amount may vary based on factors including, but not limited to, individual proficiency, skills, experience, and location.

We are an equal opportunity employer. US Citizenship may be required for certain project assignments involving security clearance.

Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Source: Code Metal careers (Ashby)

Similar roles