Software Engineer - Inference Performance
Toronto, ON +1Baseten
Candidates need a bachelor's, master's or PhD in a technical field, experience with Python or C++, and familiarity with LLM optimization and ML libraries such as PyTorch or TensorRT. They also need a deep understanding of GPU architecture and demonstrated interest in LLMs.
PyTorch, TensorRT, TensorRT-LLM, vLLM, SGLang, CUDA
Software Engineer - Model Performance Systems
Toronto, ON +1Baseten
Candidates need Python skills, interest in GPU and networking systems, and an automation mindset. Strong communication, the ability to work with vague requirements and mentoring experience are also expected. C++ is a plus.
PyTorch Profiler, NVIDIA Nsight Systems, py-spy, GitHub Codespaces, InferenceMAX, genai-bench
Software Engineer - GPU Kernels
Toronto, ON +1Baseten
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