Drimva

Baidu USA

Machine Learning System Software Architect

Sunnyvale,CA +1 · Full-time · Posted Jun 16, 2021

Location
Sunnyvale,CA
Sunnyvale, CA, United States
Experience
5+ years.
Education
MS or PhD in Electrical or Computer Engineering.

About the job

Architects high-performance machine-learning system software and distributed AI training systems for Baidu’s Kunlun AI SoC roadmap.

What you'll do

  • Architect, simulate, and design machine-learning systems, evaluating performance, power, and cost.
  • Build tools for modeling and performance evaluation, and optimize the hardware/software stack with engineering teams.
  • Track AI trends and drive partnerships to support technology and roadmap decisions.

What they ask for

  • 5+ years architecting software solutions for machine learning, acceleration, and optimization, especially distributed training or HPC.
  • Experience with deep-learning frameworks such as TensorFlow, PyTorch, or PaddlePaddle.
  • Experience with CPUs, GPUs, memory systems, accelerators, and C++ performance simulation and modeling.
  • MS or PhD in Electrical or Computer Engineering; excellent communication skills in English and Chinese.

Nice to have

  • Strong outreach track record with ML researchers and application developers.

Tools

Kunlun Accelerators, TensorFlow, PyTorch, PaddlePaddle, C++, CPUs, GPUs

AI focus

Architects machine-learning systems and distributed AI training, and advances AI accelerator hardware/software performance.

Model-written summary · the original posting is authoritative

First seen Oct 27, 2026 · Last verified Nov 5, 2026 (UTC)