Machine Learning System Hardware Architect

Baidu USA; Sunnyvale (CA); On-site; USD 150,000 - 180,000; Full time

Job summary

Baidu USA is seeking a Machine Learning System Hardware Architect to join our Sunnyvale team. This role involves architecting high-performance machine learning solutions and contributing to innovative SoC designs. The ideal candidate should have extensive experience in hardware solutions and deep learning frameworks.

Outstanding collaboration skills and a drive for excellence are a must. Join us to push the boundaries of AI technology!

Qualifications

• At least 5 years of experience architecting hardware solutions for Machine Learning.
• Experience with optimizing area, frequency, and power.
• Strong outreach to ML researchers and application developers.

Responsibilities

• Architect high-performance machine learning silicon and systems.
• Develop innovations for Baidu’s AI SoC roadmap.
• Collaborate to optimize hardware and software performance.

Skills

Architecting hardware solutions for Machine Learning

Experience with deep learning frameworks

Performance simulation and modeling in C++

Knowledge of Machine Learning market trends

Communication skills in English and Chinese

Education

MS or PhD in Electrical or Computer Engineering

Tools

TensorFlow

PyTorch

PaddlePaddle

Job description

Machine Learning System Hardware Architect

Sunnyvale, CA

Do you want to be part of the AI revolution? Do you want to think outside the box, thrive on challenges in the AI industry, and solve them? Do you want to work with a world‑class team to explore the fast‑growing AI hardware opportunities and impact on the industry?

We’re looking forward to you joining us to collaborate, contribute, and revolutionize AI silicon and systems.

Description

We are looking for a world‑class Machine Learning System Architect (HW) to join our SoC team at Baidu’s Sunnyvale office. The successful candidate will be a motivated self‑starter who will thrive in a highly technical environment. Your job responsibilities as a Machine Learning System Architect include:

• Architecting and creating high‑performance machine learning silicon and connecting thousands of Kunlun Accelerators together for distributed AI training tasks.
• Developing differentiated architectural innovations for Baidu’s Kunlun AI SoC roadmap.
• Architecting, simulating, and designing advanced machine learning solutions for our AI products.
• Building system‑level ML architectures that push the boundaries of performance, power, and latency; collaborating closely with teammates to optimize hardware and software for maximum performance.
• Monitoring industrial and academic trends in artificial intelligence and determining how they intersect with our roadmaps; driving partnerships for access to the most advanced AI technologies.
• Evaluating the power, performance, and cost of prospective architecture and subsystems; building scalable tools for modeling and performance evaluation.
• Engaging with system and application software engineers to ensure optimization of the entire hardware/software stack.
• Engaging with SoC design, verification, and validation engineers to realize the architecture.

Qualifications

• Knowledge of the Machine Learning market, technological and business trends, software ecosystem, and emerging applications.
• Proven track record of at least 5 years architecting hardware solutions for Machine Learning, acceleration, and optimization.
• Experience with deep learning frameworks including TensorFlow, PyTorch, PaddlePaddle, etc.
• Strong track record of outreach to ML researchers and application developers.
• Experience with CPUs, GPUs, memory systems, and accelerators.
• Experience with performance simulation and modeling in C++.
• Experience with SoC interconnects and NoCs.
• Experience with area, frequency, and power optimizations.
• Familiarity with video, DSP, Ethernet, and PCIe.
• MS or PhD in Electrical or Computer Engineering.
• Excellent communication skills in both English and Chinese.

Culture Fit

• Mission alignment: If you want to be part of a team to accomplish this great mission, we will provide you the best possible platform to do that.
• Self‑directed: We work best with people who are driven, motivated, and aspire to greatness.
• Hungry to learn: We are eager to see you learn new skills and grow.
• Team orientation: We work in small, fast‑moving teams. We look out for each other and pursue big goals together.

As set forth in Baidu USA’s Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.