company_site
Software Engineer, ML Accelerators, DeepMind
Google
Mountain View, USA
Junior · 2+ years experience
USD 174000-252000 / year
Discovered 2 weeks ago
cplusplusjaxpythonpytorch
Free
Job Fit Check
Base Career helps you apply smarter for this job.
?%
Ready to ScanKey skills for this role
cplusplusjaxpython
Key Skills for This Role
cplusplusjaxpythonpytorch
Full Job Posting
Responsibilities
- Direct full-stack Software (SW) role, focusing on ML compiler and inference infrastructure.
- Design and develop ML compiler stack to bridge AI workloads and low-level Hardware (HW) operators.
- Design and develop SW infrastructure to enable high performance inference.
- Drive SW/HW codesign, kernel optimization, performance tuning, and debugging.
- - Direct full-stack Software (SW) role, focusing on ML compiler and inference infrastructure. - Design and develop ML compiler stack to bridge AI workloads and low-level Hardware (HW) operators. - Design and develop SW infrastructure to enable high performance inference. - Drive SW/HW codesign, kernel optimization, performance tuning, and debugging.
Minimum qualifications:
Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience.
Experience in software development using Python and C++.
Experience with machine learning (ML) hardware accelerators, ML compiler stacks (e.g., JAX, PyTorch, XLA), and kernel development.
Preferred qualifications:
PhD degree in Computer Engineering, Computer Science, or a related field.
2 years of experience with full-stack system development for hardware accelerators.
2 years of experience with collaborative, hyper-scalar software development and engineering best practices.
Experience scoping, planning, and executing projects towards team goals in ambiguous environments.
Qualifications
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
More jobs at Google
Software Engineer, Robotics, Infrastructure Quality and Productivity, DeepMind
London, GBR
Junior
Discovered TodayView →
Director, Trust and Safety, Trusted Experiences, Google DeepMind
Sunnyvale, USA
Director
Discovered TodayView →
Director, 1P Direct to Consumer Commerce and Performance Marketing
Mountain View, USA
Director
Discovered TodayView →
Senior Software Engineer, AI/ML GenAI, Google Cloud Data Management
Sunnyvale, USA
Senior
Discovered TodayView →
Software Engineer, Robotics, Infrastructure Quality and Productivity, DeepMind
London, GBR
Discovered Today
Partnerships Manager
Bengaluru, IND
Discovered Today
Director, Regulatory Affairs, Content Advisory
London, GBR
Discovered Today
Mixed-Methods UX Researcher
San Bruno, USA
Discovered Today
Director, Trust and Safety, Trusted Experiences, Google DeepMind
Sunnyvale, USA
Discovered Today
Director, 1P Direct to Consumer Commerce and Performance Marketing
Mountain View, USA
Discovered Today
Senior Software Engineer, AI/ML GenAI, Google Cloud Data Management
Sunnyvale, USA
Discovered Today
Research Scientist, Google Research
Mountain View, USA
Discovered Today