Jobs in the United States

25+ jobs

Save this search

Added just now
Staff Software Development Engineer
Trellix
Jefferson City, MO
Join our innovative team at Trellix, where you'll lead the design and development of our high-performance Cloud Threat Detection Platform. In this role, you will lead the architecture and…
Added 1 min ago
B
Mid-Level Red Team Operator
B&A
Washington, DC
- $135,000 to $150,000 USD / year
Lead mission-critical Red Team cybersecurity exercises that comprehensively test and fortify production IT infrastructure, physical facilities, and personnel within federal court systems,…
Added 1 min ago
AI & Machine Learning Engineering Consultant - Manager - Consulting - Location OPEN
EY
Kansas City, MO
+ Ability to lead teams effectively and manage change within the organization.
Added 1 min ago
AI & Machine Learning Engineering Consultant - Manager - Consulting - Location OPEN
EY
Saint Louis, MO
+ Ability to lead teams effectively and manage change within the organization.
Added 1 min ago
AI & Machine Learning Engineering Consultant - Manager - Consulting - Location OPEN
EY
Jefferson City, MO
+ Ability to lead teams effectively and manage change within the organization.
Added 2 mins ago
Lead Software Engineer - R&D Productivity Tools
Schweitzer Engineering Laboratories, Inc.
Pullman, WA
…rapidly create and scale high-quality, custom solutions. As a Lead Software Engineer joining a new productivity tools team, you'll improve existing tools, shape the roadmap for new ones,…
Added 2 mins ago
Sr Data Scientist - Data Science & Insights
The Hertz Corporation
Jefferson City, MO
+ Lead the end-to-end model lifecycle, from problem framing and EDA to feature engineering, model tuning, deployment, and monitoring.
Added 5 mins ago
Principal FP&A and Business Intelligence
Northwest Center
Seattle, WA
Lead the division's annual planning and forecasting processes end to end, establishing clear timelines, defined inputs, and accountability for delivering accurate and timely financial plans.
Added 6 mins ago
Software Engineer II
Pinterest
Seattle, WA
…scalable and efficient manner. Actively work to improve the developer process and experience in all phases from coding to operation. Contribute to and lead each step of the product…
Added 8 mins ago
P
Principal Application Engineer (Salesforce)
PointClickCare
Remote, United States
- $179,300 to $224,100 USD / year
PointClickCare is searching for a Principal Application Engineer who will lead application development by writing and rigorously defending production-grade application code and…
Added 9 mins ago
Staff Full-stack Engineer - - B2B Platforms & Payment Systems (Remote-Eligible)
Capital One
Jefferson City, MO
…where the ideas of others can be heard and championed. You will lead the way in creating next-generation talent for Capital One Tech, mentoring engineers and actively recruiting to keep…
Added 9 mins ago
Manager, AI Risk & Validation
Deloitte
Kansas City, MO
+ 7+ years of experience developing, validating, or deploying AI, generative AI, or agentic AI solutions using Python and JavaScript
Added 10 mins ago
CH
Data & Analytics Platform
Cambia Health Solutions
Renton, WA
- $114,000 USD / year
…seeking a Senior Data Engineer to join the team as a technical lead, helping shape and deliver scalable platform capabilities that enable data engineering teams across Cambia. In this role,…
Added 11 mins ago
Chief Engineer - Dimensional Engineering
Janicki Industries Inc.
Hamilton, WA
Lead resolution of complex dimensional issues affecting design, manufacturing, tooling, assembly, and inspection.
Added 11 mins ago
AS
Commodity Manager - 3054743
Apex Systems, Inc.
Bellevue, WA
- $68 to $78 USD / hour
Lead cross-functional meetings with clear document writing, agenda creation, and follow-up activity coordination.
Added 12 mins ago
SAP CPQ Senior Consultant - AI-Driven Configuration & Commerce
Deloitte
Kansas City, MO
…capabilities that remove friction from the Quote-To-Order and Lead-To-Cash processes and replace manual exception-handling with automated, rules- and AI-driven…
Added 12 mins ago
SAP CPQ Senior Consultant - AI-Driven Configuration & Commerce
Deloitte
Saint Louis, MO
…capabilities that remove friction from the Quote-To-Order and Lead-To-Cash processes and replace manual exception-handling with automated, rules- and AI-driven…
Added 12 mins ago
Lead, Software Engineer - Backend /Integration - TS/SCI
L3Harris
Patrick Air Force Base, FL
Job Title: Lead, Software Engineer - Backend/Integration
Added 12 mins ago
Lead Data Engineer
NextEra Energy
Juno Beach, FL
We are seeking a hands-on Lead Data Engineer to architect and deliver enterprise-grade data and AI products that power an executive intelligence platform. The role sits within Marketing &…
Added 13 mins ago
Staff Perception Scientist
ARM
Seattle, WA
Lead research reviews, mentor scientists and engineers, and guide the technical roadmap for on-device perception.
Added 13 mins ago
Staff Robotics Engineer
ARM
Seattle, WA
You will provide technical leadership across robotics architecture, platform integration, and developer workflows. This is a high-impact role at the intersection of robotics, embedded…
Added 13 mins ago
A
Staff Software Engineer, AI Inference Runtime
ARM
Seattle, WA
- $209,100 to $282,900 USD / year
You will lead hands-on work across scheduling, batching, KV-cache management, memory allocation, distributed workload execution, kernel development and optimization, and performance…
Added 13 mins ago
A
Staff Software Engineer, AI Inference Cloud
ARM
Seattle, WA
- $209,100 to $282,900 USD / year
Lead production readiness reviews and resolve complex issues across services, Kubernetes, networking, and compute infrastructure. Turn incidents and operational bottlenecks into durable…
Added 13 mins ago
Senior Engineer, Process Data Science and Statistics
AstraZeneca
Gaithersburg, MD · Gaithersburg
This position reports to the Director, Process Data Science and Statistics lead, Process Engineering, Cell Therapy Development and Operations, and is located Required Qualifications
Added 14 mins ago
Consumer Intelligence & Strategic Analytics Lead, SVP - US Consumer Cards
Citi
SCHAUMBURG, IL · ATLANTA, GA · Schaumburg, IL · Atlanta, GA · Schaumburg · Atlanta
The Consumer Intelligence & Strategic Analytics Lead serves as a trusted advisor to senior leadership, delivering insights into consumer behavior, customer trends, competitive dynamics, and…

ARM

Staff Software Engineer, AI Inference Runtime

Seattle, WA, United States

$209,100 to $282,900 USD / year

Apply About employer

Applications open in a new tab.

Job details

Pay: $209,100 to $282,900 USD / year
Apply by: Nov 8, 2026
Experience: 5+ years of experience, or equivalent demonstrated impact, in ML systems, high-performance systems, compilers, kernel development, or production AI inference.
Posting language: English
Posted: Oct 8, 2026

Original job description

As a Software Engineer on our AI Inference Runtime team, you will set technical direction for critical components of distributed Inference runtime for running SOTA AI Models.

You will lead hands-on work across scheduling, batching, KV-cache management, memory allocation, distributed workload execution, kernel development and optimization, and performance benchmarking and analysis. Your work will directly influence how efficiently new models use available compute. Partnering with our AI Infrastructure, compute, and product teams to enhance the performance and efficiency of Arm's AI platform.

Responsibilities:

Define the architecture, interfaces, and roadmap for AI inference runtime capabilities, including abstractions that support evolving models, workloads, and compute platforms.
Enable new model architectures end to end through operator support, production validation, and optimization of scheduling, batching, model execution, memory management, and KV-cache efficiency.
Profile system bottlenecks and develop optimized kernels and data-movement paths across compute, memory, networking, and framework integration.
Evaluate new inference techniques and build benchmarking, regression, validation, and safe-rollout systems to improve latency, throughput, reliability, and resource efficiency.
Partner with cloud, framework, compiler, hardware, and research teams; lead technical reviews, mentor engineers, and establish meticulous performance-engineering practices.

Required Skills and Experience :

5+ years of experience, or equivalent demonstrated impact, in ML systems, high-performance systems, compilers, kernel development, or production AI inference.
Deep understanding of modern AI inference, including model execution, Attention, MoE, batching, prioritisation, and KV-cache behavior.
Strong programming skills in C++, Rust, Python, or a comparable language, with knowledge of concurrency, parallel programming, handling of memory resources, and data movement.
Proven ability to profile, debug, and optimize performance across kernels, runtimes, frameworks, operating systems, and hardware.

"Nice To Have" Skills and Experience :

Experience developing or modifying inference schedulers, cache managers, batching systems, disaggregated or distributed execution paths.
Experience optimizing kernels using accelerator programming tools, assembly, or intrinsics, including attention, matrix multiplication, operator fusion, and low-precision execution.
Familiarity with model parallelism, collective communication, high-performance networking, compilers, or graph optimization.
Contributions to open-source ML runtimes, frameworks, compilers, or kernel libraries.

In Return:

You will be part of our AI Platforms team - A driven and diverse group passionate about developing foundational production capabilities to support AI inference at Arm. We provide a collaborative setting where your ideas can come to life quickly. The success of our AI projects will be directly influenced by your work, crafting the company's AI inference capabilities and defining and operating production inference workloads. This is an outstanding opportunity to work with world-class teams and contribute to groundbreaking advances in AI technology. Join us in building the next generation of AI inference infrastructure!

Salary Range:$209,100-$282,900 per year We value people as individuals and our dedication is to reward people competitively and equitably for the work they do and the skills and experience they bring to Arm. Salary is only one component of Arm's offering. The total reward package will be shared with candidates during the recruitment and selection process.

Our 10x Mindset At Arm, we believe progress happens To view the full job description, <a href="https://www.aplitrak.com/?adid=YmJnZW5lcmljLjA0MzE5Ljg3NjlAYXJtY29tcC5hcGxpdHJhay5jb20">click here</>

Apply About employer

More about this job

Skills

Scheduling
ML
C++
Rust
Python
Assembly
Communication skills

Report this job