AI Compute Engineer — Accelerators Co-Design

Google; United States; On-site; USD 174,000 - 252,000; Full time

Job summary

Google DeepMind is building a next-generation compute platform for AI inference workloads, focusing on memory technologies, hardware acceleration, and software/hardware codesign. The role sits in Mountain View/San Francisco and offers meaningful impact on world-scale AI systems.

As a Software Engineer, you will work with scientists and engineers to advance ML hardware and software co-design, optimizing kernels and benchmarks, and shaping the technology roadmap.

Qualifications

• Bachelor’s degree in CS or related field or equivalent practical experience.
• Experience in software development using Python and C++.
• Experience with ML hardware accelerators and ML inference software.

Responsibilities

• Collaborate with product teams to design, implement, and prototype novel software algorithms.
• Oversee software/hardware codesign, kernel optimization, performance benchmarking and optimization.
• Collect insights to influence future software and hardware design and the technology roadmap.

Skills

Python

C++

ML

Education

Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience.

Job description

Google DeepMind is building a next-generation compute platform for AI inference workloads, focusing on memory technologies, hardware acceleration, and software/hardware codesign. The role sits in Mountain View/San Francisco and offers meaningful impact on world-scale AI systems.

As a Software Engineer, you will work with scientists and engineers to advance ML hardware and software co-design, optimizing kernels and benchmarks, and shaping the technology roadmap.