logo

baseten

Software Engineer - GPU Kernels

Department
Engineering
Job Type / Location
remote
Experience Required
3+ years
Posted On

We are seeking a skilled Software Engineer to join our team and contribute to the development of high-performance GPU kernels. The ideal candidate will have expertise in Python, C++, and GPU programming frameworks such as CUDA and OpenCL. As a Software Engineer - GPU Kernels, you will design and develop optimized GPU kernels to accelerate computational tasks, collaborate with cross-functional teams, and ensure high-quality code delivery.

Key Responsibilities:

  • Design and develop high-performance GPU kernels using Python and C++.
  • Collaborate with cross-functional teams, including software engineers, researchers, and data scientists.
  • Optimize GPU kernels for performance, power efficiency, and scalability.
  • Develop and maintain high-quality, well-documented code.
  • Participate in code reviews and contribute to the improvement of the codebase.

Requirements:

  • 5+ years of experience in software development, with a focus on GPU programming.
  • Expertise in Python, C++, and GPU programming frameworks such as CUDA and OpenCL.
  • Strong understanding of computer architecture, parallel processing, and optimization techniques.
  • Excellent problem-solving skills, with the ability to analyze complex problems and develop creative solutions.
  • Strong communication and collaboration skills, with the ability to work effectively with cross-functional teams.

View Assessment Process

Think you'll be a good fit?