Systems Research Engineer, GPU Programming

Job summary

Responsibilities

As a Systems Research Engineer specialized in GPU Programming, you will play a crucial role in developing and optimizing GPU-accelerated kernels and algorithms for ML/AI applications.

• Optimize and fine-tune GPU code to achieve better performance and scalability
• Collaborate with cross-functional teams to integrate GPU-accelerated solutions into existing software systems
• Stay up-to-date with the latest advancements in GPU programming techniques and technologies

What they're looking for

• Strong background in GPU programming and parallel computing, such as CUDA and/or Triton.
• Knowledge of ML/AI applications and models
• Knowledge of performance profiling and optimization tools for GPU programming
• Excellent problem-solving and analytical skills
• Bachelor's, Master's, or Ph.D. degree in Computer Science, Electrical Engineering, or equivalent practical experiences

Tech stack

• CUDA
• Triton