Staff Machine Learning Engineer

Job overview

The Staff Machine Learning Engineer role is onsite in San Jose, CA. Compensation is not specified. The role supports Axiado, an AI-enhanced security processor company developing silicon-rooted security and management chips for AI data center infrastructure. The engineer works across models, systems software, and silicon constraints to bring machine-learning capabilities into production hardware. The role contributes by optimizing training and inference, adapting models and runtimes to chip limitations, and collaborating with hardware, firmware, and QA teams through deployment and monitoring.

What you'll do

• Optimize training and inference across GPU and AI-accelerator infrastructure, including MLOps pipelines
• design, train, and evaluate machine-learning models and take them into production
• extend NPU cores into production silicon
• optimize inference engines and serving runtimes for latency, memory, and power constraints
• work on BMC firmware, embedded Linux, or RTOS as needed
• build automated test and verification harnesses
• apply machine learning to security
• collaborate with RTL/hardware, firmware, and QA teams.

What we're looking for

Skills & competencies

staff machine learning engineer

full-time

san jose ca

axiado

ai/ml

ai accelerators

npu

inference optimization

mlops

embedded linux

bmc firmware

ai security

master’s required

compensation not specified

Work arrangement

• Weekend coverage required

Why this role

Works across the full stack from model development and inference systems to chip-aware implementation; the posting requires a master’s degree and prefers a PhD.

About the employer

Axiado is hiring for this role. Industry: Semiconductor and Related Device Manufacturing. Sector: 33.