Principal CUDA Software Engineer (AI/ML)

AI Startup in Stealth
CA, United States
Full-time

Job Description :

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.

Key Responsibilities :

  • 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.

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.

Application Process :

If you are passionate about AI / ML software development and hardware optimization and want to work on groundbreaking technology, we’d love to hear from you.

Please submit your resume.

1 day ago
Related jobs
Promoted
VirtualVocations
Oakland, California

A company is looking for a Principal AI/ML Infrastructure and Ops Engineer to manage large-scale AI/ML platforms and ensure operational excellence. Key Responsibilities:Lead the design and implementation of scalable infrastructure solutionsOversee management of multi-cloud and hybrid infrastructure ...

Promoted
Raytheon
Compton, California

Experience with: System Safety Engineering, System Engineering, Software or Firmware Engineering, Electrical Engineering or equivalent Safety Engineering. Typically Requires:  Bachelor's degree in Science, Technology, Engineering or Math (STEM) and a minimum 8 years of relevant Software Engineering ...

Promoted
Blue Origin
Los Angeles, California

Mentor junior AI engineers and actively participate in the AI/ML community. Autonomous Vehicle AI Principal Engineer with experienced and demonstrated skills in advanced neural networks. Work collaboratively with a multidisciplinary team of engineers, researchers, and scientists to integrate AI/ML s...

Promoted
Autodesk
San Francisco, California

Autodesk is hiring a Senior Principal Software Engineer and Architect who has experience building (design + development) modern cloud services. Conduct code reviews to maintain high code quality standards, and mentor team members on best practices. Design a new cloud-based software system for data a...

Promoted
f5
San Jose, California

We are looking for a Sr Principal Software Engineer or Software Architect to join our Data engineering team to accelerate data platform and data and telemetry missions in the F5 Distributed Cloud Platform Group. Experience with data driven, ML/AI, Generative AI powered applications in security, netw...

Promoted
Walmart
Sunnyvale, California

Option 1: Bachelor's degree in computer science, computer engineering, computer information systems, software engineering, or related area and 5 years’ experience in software engineering or related area. Master’s degree in computer science, computer engineering, computer information systems, softwar...

Promoted
Sonance
San Clemente, California

As a Senior/Lead Embedded Systems Software Engineer, you will design and build entirely new audio products, from your strong background in both audio hardware and software. Principal Software Engineer; Audio Systems preferred. The ideal candidate can write world-class software solutions for Linux an...

Promoted
Roblox
San Mateo, California

As a Principal / Senior Software Engineer in the Observability team, you will:. Passion for mentoring junior engineers, with the ability to earn their respect and help them stretch themselves to build great software and level up their technical knowledge. We provide the vital signs for our software ...

Shield AI
San Diego, California

You’ll collaborate cross-functionally with other Core Product engineers, Solutions engineers and new customers to develop robust interfaces and architectures that enable seamless integration with new systems and user interfaces. In pursuit of this mission, Shield AI is building the world’s best AI p...

Northrop Grumman
San Diego, California

Preferred Qualifications:* * Experience managing hardware and Software related issues with a supplier-based interaction as part of a Government/DOD contract * Familiarity with full hardware integration and Software Development Life Cycles * Experience with software and f development, integration, an...