In this role, you will apply your expertise as a GPU programmer to design GPU related tasks to train LLMs, focusing on performance and architecture.
GPU Programmer AI Training Role
Responsibilities
Design, implement, and optimize GPU-based software solutions using CUDA, WebGPU or GLSL.
Profile and fine-tune GPU kernels and shaders for maximum performance and efficiency.
Develop host-side logic and integrations in C++
Qualifications
Advanced experience with any GPU programming language that can run on NVIDIA chips.
Background in either Graphics Programming, ML acceleration, Scientific Computing, High Performance Computing, or any other domain using GPU technologies.
