Senior ML Engineer — AI Silicon & Systems

Axiado Corporation; San Jose (CA); On-site; USD 150,000 - 230,000; Full time

Job summary

Axiado is seeking an ML engineer to work across the full stack from model to silicon. You will optimize training and inference on GPU/AI-accelerator infrastructure, adapt models to chip constraints, and bring AI features from training to production.

You will design and evaluate models, Harden NPU cores, and collaborate with RTL/hardware, firmware, and QA teams to ship AI features end-to-end with real hardware constraints.

Qualifications

• 5–7+ years of hands-on AI/ML experience; Master’s degree required, PhD preferred.

Responsibilities

• Optimize training and inference performance on GPU/AI-accelerator infrastructure.
• Design, train, and evaluate ML models and take them into production.
• Harden and extend NPU cores into production silicon.
• Build or optimize inference engines and serving runtimes.
• Collaborate with RTL/hardware, firmware, and QA teams to ship AI features end-to-end.

Skills

AI/ML experience

Model development

Production deployment

Chip-aware ML

GPU/accelerator optimization

Education

Master's degree

PhD preferred

Tools

GPU clusters

Inference engines

MLOps pipelines

Embedded Linux

Job description

Axiado is seeking an ML engineer to work across the full stack from model to silicon. You will optimize training and inference on GPU/AI-accelerator infrastructure, adapt models to chip constraints, and bring AI features from training to production.

You will design and evaluate models, Harden NPU cores, and collaborate with RTL/hardware, firmware, and QA teams to ship AI features end-to-end with real hardware constraints.