Software Engineer, GPU Performance
Toronto, ONHeyGen
Candidates need experience optimizing GPU-based AI or high-performance computing systems, strong Python and PyTorch skills, and knowledge of GPU hardware. Using profiling tools to validate improvements is required. CUDA, Triton and video or diffusion model experience are preferred.
PyTorch, NVIDIA Nsight Systems, NVIDIA Nsight Compute, PyTorch Profiler, CUDA, Triton
Data Infrastructure Engineer
Toronto, ONHeyGen
Candidates need a Bachelor's or Master's in computer science, engineering or a related field, plus proven experience running infrastructure for large-scale AI or machine learning projects. Strong Python skills and familiarity with TensorFlow or PyTorch are required.
Python, TensorFlow, PyTorch