We are seeking a highly skilled and experienced Principal CUDA 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.
Lead the development of CUDA-based 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.
Lead CUDA performance analysis, benchmarking, and profiling.
Bachelor's or master's degree in computer science, electrical engineering, or a related field.
10+ years of experience in software development.
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.
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.