Machine Learning System Software Architect

Baidu USA; Sunnyvale (CA); On-site; USD 130,000 - 160,000; Full time

Job summary

Baidu USA is seeking a Machine Learning System Software Architect in Sunnyvale, CA. You'll architect high-performance machine learning software solutions and lead the creation of distributed AI systems.

The ideal candidate has 5+ years of experience, deep knowledge of ML trends, and expertise in deep learning frameworks. Join us and become a key player in the AI revolution.

Expect a collaborative environment focused on innovation and excellence.

Qualifications

• 5+ years architecting software solutions for Machine Learning and distributed training systems.
• Excellent communication skills in English and Chinese.

Responsibilities

• Architect and create high-performance machine learning system software.
• Design machine learning solutions for AI products.
• Collaborate with other teams to optimize hardware/software for performance.

Skills

Machine Learning market knowledge

Software solutions architecture

Deep learning frameworks (TensorFlow, PyTorch, PaddlePaddle)

Outreach to ML researchers

Performance simulation and modeling in C++

CPUs, GPUs, memory systems, and accelerators knowledge

Education

MS or PhD in Electrical or Computer Engineering

Job description

Machine Learning System Software Architect

Sunnyvale, CA

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.

Responsibilities

As a Machine Learning System Software Architect, you 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. Your responsibilities include:

• 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.

Equal Employment Opportunity

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.