Hybrid Senior Deep Learning Software Engineer, PyTorch - TensorRT Performance

Posted last month

Apply now

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

Job title

Senior Deep Learning Software Engineer, PyTorch - TensorRT Performance

Job type

Experience level

Senior

Salary

$148,000 - $287,500 per year

Degree requirement

Bachelor's Degree

Location requirements

Report this job

See something inaccurate? Let us know and we'll update the listing.

Report job