Be an early applicantLess than 25 applicants

Company

NVIDIA · 14 hours ago

Senior AI Platform Engineer

US, OR, Remote

Full-time

Remote

Senior Level, Lead/Staff

$180K/yr - $339K/yr

8+ years exp

Maximize your interview chances

Artificial Intelligence (AI)GPU

Growth Opportunities

H1B Sponsor Likely

Insider Connection @NVIDIA

Discover valuable connections within the company who might provide insights and potential referrals.
Get 3x more responses when you reach out via email instead of LinkedIn.

Responsibilities

Develop and refine infrastructure for scalable AI workloads optimized specifically for NVIDIA hardware and software.

Debug and troubleshoot AI jobs running on NVIDIA hardware, leveraging tools such as Nsight, CUDA, and NCCL, to enhance efficiency and performance.

Collaborate with internal teams, including research, engineering, and data science, to identify areas for efficiency improvement across NVIDIA's AI workloads.

Optimize AI frameworks, particularly PyTorch, to fully utilize NVIDIA hardware capabilities.

Create and maintain NVIDIA-specific performance analysis tools and scripts to monitor, assess, and improve AI infrastructure reliability and performance.

Utilize NVIDIA's profiling tools (e.g., Nsight Systems, Nsight Compute, DCGM) to gain deep insights into bottlenecks and optimize AI task performance on NVIDIA hardware.

Qualification

Find out how your skills align with this job's requirements. If anything seems off, you can easily click on the tags to select or unselect skills to reflect your actual expertise.

AI workloadsPyTorchC++NVIDIA GPUsCUDANsightNCCLGPU programmingNVIDIA hardware architecturesPerformance tuningDistributed AI systemsSlurm workload managerK8sParallel programmingMulti-threaded computing

Required

8+ years of experience in debugging and optimizing AI workloads on NVIDIA GPUs.

Bachelor's or Master's degree in Computer Science, Software Engineering or equivalent experience.

Strong programming experience with Python, especially using AI frameworks like PyTorch, with a focus on NVIDIA optimization.

Proficiency in C++, with experience in GPU programming, CUDA, and familiarity with NVIDIA-specific optimizations.

Knowledge of NVIDIA hardware architectures, including GPU memory hierarchy, CUDA cores, and performance tuning techniques.

Preferred

Experience with large-scale distributed AI systems on cloud or on-premises infrastructure using NVIDIA hardware.

Extensive experience in Slurm workload manager and K8s.

Understanding of parallel programming and experience with multi-threaded or distributed computing on NVIDIA platforms.

Benefits

Equity

Benefits

Company

NVIDIA

Glassdoor

4.6

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.

Founded in 1993

Santa Clara, California, USA

10,001+ employees

https://www.nvidia.com

H1B Sponsorship

NVIDIA has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2023 (735)

2022 (892)

2021 (696)

2020 (534)