AI Accelerator Engineer (HW/SW Co-Design) – Equity

Socket.dev; Mountain View (CA); On-site; USD 174,000 - 252,000; Full time

Benefits offered by this job

Bonus target 15%

Job summary

Google DeepMind is building a next-generation compute platform for AI inference workloads, with emphasis on memory technologies, hardware acceleration, and codesign. You’ll join scientists and engineers shaping the future of machine learning for Google and industry.

The role focuses on collaboration, kernel optimization, performance benchmarking, and guiding the technology roadmap with practical insights across software and hardware domains.

Qualifications

• Bachelor's degree or equivalent in a technical field with practical experience.
• Experience developing software in 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 roadmap.

Skills

Python

C++

ML hardware accelerators

Education

Bachelor's degree in Computer Science or related field

PhD in Computer Engineering/Computer Science or related field

Job description

Google DeepMind is building a next-generation compute platform for AI inference workloads, with emphasis on memory technologies, hardware acceleration, and codesign. You’ll join scientists and engineers shaping the future of machine learning for Google and industry.

The role focuses on collaboration, kernel optimization, performance benchmarking, and guiding the technology roadmap with practical insights across software and hardware domains.