{"version":"0.1","company":{"name":"YubHub","url":"https://yubhub.co","jobsUrl":"https://yubhub.co/jobs/skill/nvidia-ai-stack"},"x-facet":{"type":"skill","slug":"nvidia-ai-stack","display":"Nvidia Ai Stack","count":1},"x-feed-size-limit":100,"x-feed-sort":"enriched_at desc","x-feed-notice":"This feed contains at most 100 jobs (the most recently enriched). For the full corpus, use the paginated /stats/by-facet endpoint or /search.","x-generator":"yubhub-xml-generator","x-rights":"Free to redistribute with attribution: \"Data by YubHub (https://yubhub.co)\"","x-schema":"Each entry in `jobs` follows https://schema.org/JobPosting. YubHub-native raw fields carry `x-` prefix.","jobs":[{"@context":"https://schema.org","@type":"JobPosting","identifier":{"@type":"PropertyValue","name":"YubHub","value":"job_18ab3b67-c62"},"title":"Senior Software Engineer – PyTorch and AI Frameworks","description":"<p>We are looking for experienced engineers to help build and scale next-generation AI infrastructure using PyTorch, one of the world&#39;s most widely used deep learning frameworks. This role sits at the intersection of machine learning systems, compilers, and high-performance computing, enabling researchers and product teams to train and deploy large-scale models efficiently.</p>\n<p>You will work on core components of the PyTorch ecosystem, including model execution, distributed training, performance optimization, and developer experience.</p>\n<p>Responsibilities:</p>\n<ul>\n<li>Design and build core PyTorch capabilities across runtime, autograd, distributed training, and model execution</li>\n<li>Optimize performance across GPU/accelerator backends (CUDA, Triton, etc.)</li>\n<li>Contribute to or lead development of large-scale ML systems and infrastructure</li>\n<li>Improve model training efficiency, scalability, and reliability across multi-node environments</li>\n<li>Work on compilers / graph transformations / kernel optimization to accelerate deep learning workloads</li>\n<li>Partner with researchers and applied teams to translate cutting-edge models into production systems</li>\n<li>Drive open-source contributions and collaborate with the broader PyTorch community</li>\n<li>Influence roadmap and architecture for next-gen AI platforms</li>\n<li>Work at the forefront of AI and accelerated computing</li>\n<li>Direct impact on how PyTorch runs on the world&#39;s most advanced GPU platforms</li>\n<li>Collaborate across hardware, systems software, and AI research to push performance boundaries and enable breakthroughs in generative AI, autonomous systems, and high-performance computing</li>\n</ul>\n<p>Requirements:</p>\n<ul>\n<li>PhD or MSc degree in Computer Science, Applied Math, Physics, or related science or engineering field (or equivalent experience)</li>\n<li>8+ years of software development experience</li>\n<li>Strong programming skills in C++ and Python</li>\n<li>Deep understanding of deep learning frameworks, preferably PyTorch</li>\n<li>Experience with GPU programming (CUDA or similar) and performance optimization</li>\n</ul>\n<p>Preferred qualifications:</p>\n<ul>\n<li>Contributions to PyTorch core or ecosystem libraries</li>\n<li>Experience with NVIDIA AI stack (TensorRT, Triton Inference Server, cuBLAS, cuDNN, NCCL)</li>\n<li>Familiarity with ML compilers (TorchInductor, Triton, XLA, TVM)</li>\n<li>Experience optimizing LLMs or large-scale recommendation / vision models</li>\n<li>Background working closely with hardware-aware software optimization</li>\n</ul>\n<p>Benefits:</p>\n<ul>\n<li>Competitive salaries</li>\n<li>Generous benefits package</li>\n<li>Eligible for equity</li>\n</ul>\n<p>Application deadline: April 27, 2026</p>\n<p style=\"margin-top:24px;font-size:13px;color:#666;\">XML job scraping automation by <a href=\"https://yubhub.co\">YubHub</a></p>","url":"https://yubhub.co/jobs/job_18ab3b67-c62","directApply":true,"hiringOrganization":{"@type":"Organization","name":"NVIDIA","sameAs":"https://www.nvidia.com/","logo":"https://logos.yubhub.co/nvidia.com.png"},"x-apply-url":"https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-Software-Engineer---PyTorch-and-AI-Frameworks_JR2016912","x-work-arrangement":"onsite","x-experience-level":"senior","x-job-type":"full-time","x-salary-range":null,"x-skills-required":["PyTorch","C++","Python","GPU programming","performance optimization"],"x-skills-preferred":["PyTorch core","NVIDIA AI stack","ML compilers","LLMs","large-scale recommendation / vision models"],"datePosted":"2026-04-24T12:14:28.799Z","jobLocation":{"@type":"Place","address":{"@type":"PostalAddress","addressLocality":"Santa Clara"}},"employmentType":"FULL_TIME","occupationalCategory":"Engineering","industry":"Technology","skills":"PyTorch, C++, Python, GPU programming, performance optimization, PyTorch core, NVIDIA AI stack, ML compilers, LLMs, large-scale recommendation / vision models"}]}