Senior Deep Learning Software Engineer, PyTorch - TensorRT Performance

This role is based in Santa Clara, CA or can be performed remotely.

Compensation:

$148,000 - $287,500/year

Responsibilities:

• Analyze performance issues and optimize Torch-TensorRT/TensorRT.
• Contribute features and code to NVIDIA/OSS inference frameworks.
• Collaborate with teams across generative AI, automotive, robotics, and more.
• Scale performance of deep learning models across NVIDIA accelerators.

Requirements:

• Bachelor's, Master's, PhD, or equivalent experience in relevant fields.
• At least 4 years of relevant software development experience.
• Excellent Python/C++ programming and software design skills.
• Experience with deep learning frameworks like PyTorch.
• Experience with performance analysis and optimization.

Tech stack:

PyTorch, TensorRT, CUDA, deep learning frameworks.