Software Engineer - GPU Networking & Distributed Systems
Toronto, ON +1HybrideTemps plein
Baseten
Baseten needs deep experience with high-performance networking such as InfiniBand and RoCE v2, and fluency in C++ or Python. Candidates should understand NVIDIA GPU memory architecture and be comfortable debugging low-level systems.
InfiniBand, RoCE, NCCL, NVSHMEM, NVLink, TensorRT-LLM
Software Engineer - GPU Kernels
Toronto, ON +1HybrideTemps plein
Baseten
Candidates need a strong understanding of GPU architecture and memory hierarchy, proficiency in C++ and GPU profiling tools, and knowledge of the CUDA C++ API, numerical precision and modern GPU features. Experience with Transformer models, Flash Attention, Cutlass, Triton or GEMM tuning is a plus.
CUDA, PTX, Nsight Systems, Nsight Compute, Torch Profiler, Cutlass