Hybrid Senior Deep Learning Software Engineer, PyTorch - TensorRT Performance

at NVIDIA

• Job title: Senior Deep Learning Software Engineer, PyTorch - TensorRT Performance
• Job type: Full Time
• Experience level: Senior
• Salary: $148,000 - $287,500 per year
• Degree requirement: Bachelor's Degree
• Tech skills: Python, PyTorch, Tensorflow
• Location requirements: Hybrid, Santa Clara, United States

About the role

• Senior Deep Learning Software Engineer focused on optimizing PyTorch with TensorRT for NVIDIA accelerators. Collaborating with diverse teams to enhance performance in generative AI and more.

Responsibilities

• Analyze performance issues and identify performance optimization opportunities inside Torch-TensorRT/TensorRT.
• Contribute features and code to NVIDIA/OSS inference frameworks including but not limited to Torch-TensorRT/TensorRT/PyTorch.
• Work with cross-collaborative teams inside and outside of NVIDIA across generative AI, automotive, robotics, image understanding, and speech understanding to develop innovative inference solutions.
• Scale performance of deep learning models across different architectures and types of NVIDIA accelerators.

Requirements

• Bachelors, Masters, PhD, or equivalent experience in relevant fields (Computer Science, Computer Engineering, EECS, AI).
• At least 4 years of relevant software development experience.
• Excellent Python/C++ programming, software design and software engineering skills.
• Experience with a DL framework like PyTorch, JAX, TensorFlow.
• Experience with performance analysis and performance optimization.

Benefits

• Equity
• Benefits