Senior Deep Learning Framework Communications Engineer at NVIDIA

You're a Senior Deep Learning engineer with experience in frameworks like PyTorch or JAX and a solid understanding of AI models and parallelisms. A Bachelor's degree in Computer Science or a related field, or equivalent experience, is expected. **What makes it worth a look...** This is a full-time, fully remote Senior Deep Learning Framework Communications Engineer role at Nvidia, offering an annual salary between $152,000 and $287,500 USD. You'll contribute to core communication libraries like NCCL and NVSHMEM, directly impacting AI and HPC application scaling for diverse multi-GPU demands. **You might be a good fit if you...** * Have experience integrating communication libraries into AI frameworks like PyTorch and JAX. * Can perform deep analysis of AI workloads and frameworks for multi-GPU communication. * Possess rapid prototyping skills with Python, C++, and CUDA or related DSLs. * Are familiar with performance benchmarking on AI clusters and profiler tools.