The company is looking to develop and optimize GPU-accelerated kernels and algorithms for ML/AI applications.
Requirements
- 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
Responsibilities
- 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
Other
- Bachelor's, Master's, or Ph.D. degree in Computer Science, Electrical Engineering, or equivalent practical experiences
- Excellent problem-solving and analytical skills
