Responsibilities
- Architect and create high-performance machine learning system software and distributed AI training systems connecting thousands of Kunlun Accelerators and servers.
- Develop differentiated architectural innovations for Baidu’s Kunlun AI SoC roadmap.
- Design and optimize system-level ML architectures for performance, power, and latency.
- Monitor AI industry and academic trends and determine how they should influence product roadmaps.
- Drive partnerships with ML researchers and application developers to access advanced AI technologies.
- Evaluate the power, performance, and cost of prospective architectures and subsystems.
- Build scalable tools for architecture modeling and performance evaluation.
- Collaborate with system and application software engineers to optimize the full hardware/software stack.
- Work with SoC design, verification, and validation engineers to execute the architecture.
Requirements
- Knowledge of machine learning market, technology and business trends, software ecosystems, and emerging applications.
- At least 5 years of experience architecting machine learning software solutions, acceleration, or optimization, especially for large distributed training systems and HPC environments.
- Experience with TensorFlow, PyTorch, or PaddlePaddle.
- Experience with CPUs, GPUs, memory systems, and accelerators.
- Experience with performance simulation and modeling in C++.
- A proven track record of outreach to ML researchers and application developers.
- MS or PhD in Electrical or Computer Engineering.
- Excellent communication skills in both English and Chinese.
Benefits
- The position is based at Baidu’s Sunnyvale office.
- The company describes a self-directed, learning-focused, collaborative culture with opportunities to contribute to AI silicon and system development.
Categories
About Baidu
Baidu is a leading AI company with strong Internet foundation, driven by our mission to “make the complicated world simpler through technology”. Founded in 2000 as a search engine platform, we were an early adopter of artificial intelligence in 2010. Since then, we have established a full AI stack, from deep learning frameworks to models and applications, with industry-leading self-developed technology at every level. Over the past two decades, we have built a diversified portfolio of products and services, spanning an ecosystem of hundreds of millions of users, millions of developers, and hundreds of thousands of enterprises. We have now integrated our leading AI capabilities into our offerings, driving innovative use cases. Join us to shape the future of AI and technology. For media inquiries, contact us at intlcomm@baidu.com. For career opportunities, visit https://talent.baidu.com/static/index.html