Role overview
Note: The job is a remote job and is open to candidates in USA. NVIDIA is at the forefront of the generative AI revolution, particularly focusing on optimizing generative AI models for maximal inference efficiency. They are seeking a Deep Learning Software Engineer to develop and scale up automated inference and deployment solutions, working on state-of-the-art generative AI models and enhancing NVIDIA's software platforms.
What you'll work on
Skills • Experience in Deep Learning. • Excellent software design skills, including debugging, performance analysis, and test design. • Strong proficiency in Python, PyTorch, and related ML tools (e.g. HuggingFace). • Strong algorithms and programming fundamentals. • Contributions to PyTorch, JAX, or other Machine Learning Frameworks. • Knowledge of GPU architecture and compilation stack, and capability of understanding and debugging end-to-end performance. • Familiarity with NVIDIA's deep learning SDKs such as TensorRT. • Experience in writing high-performance GPU kernels for machine learning workloads in frameworks such as CUDA, CUTLASS, or Triton.
Education Requirements • Masters, PhD, or equivalent experience in Computer Science, AI, Applied Math, or related field.
Benefits • Equity • Benefits
Company Overview • NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI. It was founded in 1993, and is headquartered in Santa Clara, California, USA, with a workforce of 10001+ employees. Its website is https://www.nvidia.com.
Company H1B Sponsorship • NVIDIA has a track record of offering H1B sponsorships, with 1418 in 2025, 1356 in 2024, 976 in 2023, 835 in 2022, 601 in 2021, 529 in 2020. Please note that this does not guarantee sponsorship for this specific role.