Page 1 of 2

Principal GPU Software Engineer (AI/ML)

📍 Mountain View, CA 🏢 In Office ⏰ Full-time

Job Description:

We are seeking a highly skilled and experienced Principal GPU Software Engineer to join our team and lead the development and optimization of GPU software for AI/ML workloads. You will work closely with hardware engineers to design and implement software components that maximize performance and efficiency. Your expertise in low-level GPU performance optimizations will be critical to ensuring that our AI/ML solutions run seamlessly and at peak performance. 

Key Responsibilities:

Lead the development of GPU software components, focusing on AI/ML workloads. 

Collaborate with hardware engineering teams to define software requirements and optimize performance at a low level. 

Develop and maintain drivers, runtime libraries, and other system software to support AI and machine learning models on GPUs. 

Implement and optimize compiler toolchains to target GPUs effectively. 

Ensure software compatibility with leading AI/ML frameworks. 

Conduct performance analysis, benchmarking, and profiling to identify and implement optimizations that leverage GPU resources effectively. 

Qualifications:

Bachelor’s or master’s degree in Computer Science, Electrical Engineering, or a related field.

10+ years of experience in software development. 

Proven experience in GPU software development (CUDA/OpenCL/HIP/ROCm/SYCL), including working with low-level hardware and performance optimization techniques. 

Proficiency in C/C++ and experience in Rust. 

Deep understanding of GPU architectures, parallel computing, and AI/ML software stacks. 

Experience with compiler development and toolchain optimization highly desirable. 

Comfortable working in a fast-paced, startup environment and taking ownership of critical projects. 

What We Offer:

A key leadership role in a rapidly growing startup shaping the future of AI. 

Competitive salary, equity options, and benefits. 

Opportunities for professional growth and development in a dynamic work environment. 

Tell us a bit more about yourself

Anything else we should know?

Upload your resume or CV

How did you hear about this job?

How did you hear about this job?
A
B
C
D
E
F
G

Are you able to work in our Mountain View, CA office or are you open to relocate to the San Francisco Bay Area?