Machine Learning System Software Architect
- Location
- Sunnyvale, CA · On-site
- Job type
- Full-time
- Level
- Lead / Staff
- Experience
- 5+ years
- Education
- Master's degree
- Listed
- 30 days ago
Skills & highlights for this role
- distributed training systems
- deep learning frameworks (TensorFlow, PyTorch, PaddlePaddle)
- performance simulation and modeling in C++
Also useful
- GPU and accelerator architecture
- HPC
About the role
Do you want to be part of the AI revolution? Do you want to think out of the box, thriving on challenges in 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 AI industry?
We're looking forward to you joining us to collaborate, contribute, and revolutionize AI silicon and system.
About the role
We are looking for a world-class Machine Learning System Software Architect 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 Software Architect will help the team to architect and create high-performance machine learning system software and build the distributed AI training system by connecting thousands of Kunlun Accelerators and servers.
- 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; and 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.
- Work with SoC design, verification, and validation engineers to execute the architecture.
Qualifications
- Knowledge of Machine Learning market, technological and business trends, software ecosystem, and emerging applications.
- Proven track record of 5+ years architecting software solutions for Machine Learning, acceleration and optimization, especially in large distributed training system and HPC area.
- Experience with deep learning frameworks: 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++
- 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
From Baidu USA's careers page, first listed on Rolev.ai on October 10, 2026. Rolev.ai is not the employer.
How Rolev.ai helps with this application
Rolev.ai is a job search and application tool for US job seekers. For this Machine Learning System Software Architect role at Baidu USA, it shows how your resume fits, tailors your resume to the posting, and prepares the application for you to review.
See how you fit
Openings from employer career sites are ranked against your whole resume—and searchable inside Claude, ChatGPT, and Cursor via Rolev.ai MCP.
Tailor without inventing
Every change to your resume is checked against your original. New experience goes in only with your yes—zero fabricated claims.
Draft the full kit
Tailored and Full Match resumes, role-specific cover letters, and answers to employer screening questions are prepared in seconds.
1-Click Extension Autofill
The Chrome extension fills the employer's Greenhouse, Workday, Lever, or Ashby form in 1 click. You review it and press Submit.
Common questions
Where is the Machine Learning System Software Architect job at Baidu USA located?
This Machine Learning System Software Architect position at Baidu USA is located in Sunnyvale, CA (On-site). Always check Baidu USA's application page for specific state eligibility or relocation policies.
What qualifications are required for Machine Learning System Software Architect at Baidu USA?
For the Machine Learning System Software Architect role, Baidu USA highlights 5+ years of relevant experience; a master's degree; experience with distributed training systems, deep learning frameworks (TensorFlow, PyTorch, PaddlePaddle), performance simulation and modeling in C++. Rolev.ai can score your existing resume against these requirements and show any skill gaps before you apply.
How do I apply for the Machine Learning System Software Architect position at Baidu USA?
This listing links directly to Baidu USA's Greenhouse application page. You can apply directly on Baidu USA's site, or use Rolev.ai to check your resume fit, generate a zero-hallucination tailored resume and cover letter for this posting, and autofill the Greenhouse application form in 1 click with the Rolev.ai Chrome extension. You always review the form and press Submit yourself.