Jobs Ask AI Pay Interviews Companies CompareSign in

Wait…

We read every job from the company's own job board. You apply on the company site.Jobs Job types Internships New graduate jobs Companies Pay Reported pay Check an offer Add your pay Interview questions Add your interview questions Compare with LinkedIn Job sites compared Pay transparency laws API for developers Monthly report Companies we cover Make a portfolio page For companies: find people About Privacy

Staff AI Platform Engineer

Code Metal, Boston Hub +3

Posted pay

$205K–$250K

a year

Location: Boston Hub +3
Work type: Hybrid
Level: Staff
Experience: 8+ years
Skills in the job text: Python
Kubernetes
PyTorch
Hugging Face
MLflow
vLLM
Posted: Oct 6, 2026
Last check: 1 h ago
Job type: ML / AI engineering, Inference & ML systems

Apply on the Code Metal site Sign in to save

The Code Metal site opens in a new tab.

CM

Code Metal

About Code Metal: funding, hiring and pay

Check my fit

Compare your résumé with this job. See where you match, what is missing and which lines to change.

Sign in to check your fit

The role

Code Metal's engineering teams are building AI-driven code transpilation, modernization, optimization, modeling and simulation solutions. They all 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 a technical lead on this team. You'll architect and build the AI enablement stack our engineers depend on, from GPU inference serving and model gateway up through agent harnesses, context engineering, observability, and AI experimentation management. It is 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 AI Research team, and you'll sometimes run experiments yourself when a platform decision needs evidence.

What you will do

Set the technical direction and architecture for Code Metal's AI platform and the team building it. Own the design docs and RFCs, help with build-vs-buy decisions, and align with other technical stakeholders.
Deploy, benchmark, and tune elastic, production inference for open-weight models on runtimes such as 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.
Build context-engineering services for memory, retrieval, and data discovery, so agents get the information specific to their tasks.
Instrument the stack end to end, for example, with OpenTelemetry traces and service metrics. Build the experiment-tracking and artifact layer that lets engineers and researchers reproduce and compare results.
Design for productization from day one (versioned APIs, security, deployment in customer environments), and communicate with business and product teams about the roadmap.

What you 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 / 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.

Experience Level

Typically 8+ years of software engineering experience, including 4+ years building and operating ML/AI or LLM systems in production, with demonstrated Staff-level scope and impact.

Nice to have

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 management 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 regulated domains such as defense or aerospace.

Pay and 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.

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

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.

Legal notes

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

Read the job posting on the Code Metal site

Similar jobs at other companies

WB
System Requirements Engineer, In-Cabin Perception
Woven by ToyotaTokyo
- Hybrid
- ML / AI engineering
Pay not posted
Posted Aug 24, 2026
WB
シニアコンピュータビジョン／機械学習エンジニア・Digital Twin Platform
Woven by Toyota東京都中央区
- Hybrid
- Infra, platform & SRE
Pay not posted
Posted Aug 25, 2026
WB
Senior Technical Program Manager, Active Learning Loop
Woven by ToyotaTokyo
- Senior
- Hybrid
- Product management
Pay not posted
Posted 16 days ago
WB
バックエンドエンジニア・Robot Platform Software
Woven by Toyota東京都中央区
- Hybrid
- Infra, platform & SRE
Pay not posted
Posted Jun 15, 2026
WB
データエンジニア・City Data Platform Development（障害者雇用）
Woven by Toyota東京都中央区
- Hybrid
- Infra, platform & SRE
Pay not posted
Posted Dec 10, 2024
WB
システム要求エンジニア・In-Cabin Perception
Woven by Toyota東京都中央区
- Hybrid
- ML / AI engineering
Pay not posted
Posted Aug 24, 2026

Source: the Code Metal Ashby job board. We last checked this job 1 h ago. We first saw it 1 h ago. A job on a job board does not always mean that the team interviews people now. We found this job board in a public list. We did not yet compare it with the careers page of the company.

Interview questions

No one has reported an interview at Code Metal yet. Did you interview there? Add the questions.

Compared with similar jobs

This base pay is below the median for staff ML and AI engineering jobs in the Bay Area.

Median: $265K a year
Other jobs in the group: 224

We compare the middle of each posted range. This is base pay as posted. The group does not include this job.

Code Metal

Open tech jobs: 23
With posted pay: 100%
Median posted base: $180K–$250K

About Code Metal: funding, hiring and pay