Why this role stands out
This hybrid role offers a unique opportunity to shape the future of AI hardware with a world-class team, driving innovation in machine learning systems for impactful AI products. You'll thrive here if you're a motivated self-starter eager to push performance boundaries and contribute to groundbreaking silicon architecture. Join us to revolutionize AI and advance your career at the forefront of technology.
Quick overview
Job description
Do you want to be part of the AI revolution? Do you want to think out of the box, thriving on challenges in the AI industry and the desire to solve them? Do you want to work with a world-class team to explore the fast-growing AI hardware opportunities and impact on the AI industry?
We’re looking forward to you joining us to collaborate, contribute, and revolutionize AI silicon and system.
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 this highly technical environment. Your job responsibilities as a Machine Learning System Architect will help the team to architect and create high-performance machine learning silicon and connect thousands of Kunlun Accelerators together for distributed AI training tasks.
Create differentiated architectural innovations for Baidu’s Kunlun AI SoC roadmap. Architect, simulate, and design amazing machine learning solutions for our AI machine learning products.
Develop system-level ML architectures that push the boundaries of performance, power, and latency; collaborate closely with many other teammates to ensure we design and optimize hardware and software for maximum performance.
Monitor industrial and academic trends in artificial intelligence and determine where they should intersect our roadmaps. Drive partnerships for access to the most advanced AI technologies
Evaluate the power, performance, and cost of prospective architecture and subsystems. Build scalable tools for modeling and performance evaluation.
Engage with system and application software engineers to ensure optimization of the entire hardware/software stack.
Engage with SoC design, verification, and validation engineers to realize the architecture.
Qualifications
- Knowledge of Machine Learning market, technological and business trends, software ecosystem, and emerging applications.
- Proven track record 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 that 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 watch out for each other and go after big goals together as a team.
#LI-DNI
Similar jobs
- TT
Member of Technical Staff, ML SystemsRecently posted Auto Apply
True Talent Solutions Partners · Menlo Park · On-site 🇺🇸
CUDADeep LearningPyTorch+1Salary not listed - RA
Senior Inference Optimization ML EngineerRecently posted Auto Apply
Rhoda Ai · Mountain View · Hybrid 🇺🇸
AWSRoboticsLLM+2TechnologySalary not listed - FP
Senior ML EngineerRecently posted Auto Apply
Flagship Pioneering, Inc. · Cambridge · Hybrid 🇺🇸
Talent AcquisitionApplicant Tracking SystemsMLOps+13TechnologySalary not listed - CL

Mid-Level Machine Learning EngineerRecently posted Auto Apply
Clera · San Francisco · On-site 🇺🇸
DockerGCPAWS+8TechnologySalary not listed - VE
Lead ML Engineer Auto Apply
Versant · Englewood Cliffs, NEW JERSEY · Hybrid 🇺🇸
GCPAWSComputer Vision+4Technology$190k - $210k/yr - ST

Senior Machine Learning Engineer–Field EngagementRecently posted Auto Apply
Striveworks · Washington · Hybrid 🇺🇸
ETLPythonKubernetes+1Technology$160k - $200k/yr - ST

Senior Machine Learning Engineer, Agentic Systems Auto Apply
Striveworks · Remote or Austin · Remote 🇺🇸
DockerRustScala+18Technology$160k - $200k/yr - ST

Senior Machine Learning Engineer, Agentic Systems (Active Secret Clearance) Auto Apply
Striveworks · Remote or Austin · Remote 🇺🇸
DockerRustScala+18Technology$185k - $230k/yr - ST

Junior Machine Learning Engineer Auto Apply
Striveworks · Austin · Hybrid 🇺🇸
DockerScikit-learnLLM+7Technology$100k - $130k/yr - ST

Senior Machine Learning Engineer (Active Secret Clearance) Auto Apply
Striveworks · Tacoma · Hybrid 🇺🇸
Scikit-learnPyTorchPython+2Technology$185k - $230k/yr - ST

Machine Learning Engineer (Active TS/SCI Clearance) Auto Apply
Striveworks · Fort Belvoir · Hybrid 🇺🇸
DockerRustScala+8Technology$140k - $190k/yr - ST

Senior Machine Learning Engineer (Active Secret Clearance) Auto Apply
Striveworks · Austin · Hybrid 🇺🇸
Scikit-learnPyTorchPython+2Technology$185k - $230k/yr