Senior Software Engineer, PyTorch - Deep Learning
NVIDIA Corporation
Senior Deep Learning Software Engineer, PyTorch Overview NVIDIA is hiring software engineers to design and build tools used by AI engineers across the world to design, develop, and deploy AI applications scalable across thousands of GPUs. This position embeds you in an ambitious and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to PyTorch, a premiere deep learning framework. Responsibilities Design and build PyTorch components that run efficiently on supercomputers with 1,000s‑100,000s of GPUs. Collaborate with NVIDIA’s hardware and software teams to improve overall GPU performance in PyTorch. Design, build and support production AI solutions used by enterprise customers and partners. Work with internal applied researchers to improve their AI tools. Required Qualifications BS in Computer Science or Engineering (or equivalent experience). 3+ years professional experience in deep learning. Proficient with C++ programming. Strong understanding of systems software and interfaces. Demonstrated experience with thread and distributed parallel programming. Demonstrated background developing large software projects. Strong verbal and written communication skills. Preferred Qualifications Contributions and participation in the open source community. Familiarity with deep learning compilers. Familiarity with deep learning modeling trends. Background with CUDA programming and Python. Demonstrated experience working with multi‑disciplinary teams. Benefits Competitive base salary ranging from 152,000USD to 241,500USD for Level3 and from 184,000USD to 287,500USD for Level4, plus equity and a comprehensive benefits package. NVIDIA is committed to fostering an inclusive work environment and is proud to be an equal‑opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law. #J-18808-Ljbffr NVIDIA Corporation
- We are now looking for a Senior Deep Learning Software Engineer, PyTorch. NVIDIA is hiring software engineers to design and build tools used by AI engineers across the world to develop scalable AI applications on thousands of GPUs. This position embeds you in an ambitious...Senior
$184k - $287.5k
...Overview NVIDIA is hiring senior engineers to develop its AI platform... ...performance optimizations in deep learning frameworks using JAX, a tool... ...and their internals (e.g., PyTorch, TensorFlow, scikit-learn,... ...numeric libraries, modular software design. Highly motivated with...Senior$153k - $222k
...Bangalore; Seoul; and Tokyo. Learn more at applied.co. We... ...both infrastructure engineers with expertise in... ...training approaches for deep learning model training... ...degree in Computer Science, Software Engineering, or... ...ML modeling frameworks (PyTorch, Tensorflow, etc.), and...SeniorFull timeFor contractorsFor subcontractorCasual workWork at officeRemote workDay shift$224k - $356.5k
We are looking for a Senior Deep Learning Software Engineer to design and build our automated inference and deployment solution. As part of the team,... ...encompassing developing features in high-level frameworks like PyTorch and JAX, designing and implementing a high-performance...Senior- We are now looking for a Senior Deep Learning Software Engineer, LLM Performance! NVIDIA is seeking an experienced Deep Learning Engineer passionate... ...skills. Experience with a deep learning framework such as PyTorch, JAX, or TensorFlow. Ways to stand out from the crowd:...Senior
$152k - $241.5k
...We are looking for a highly motivated senior software engineer for an exciting role in our... ...products in High Performance Computing and Deep Learning. What you will be doing: Design, implement... ...frameworks (e.g. NCCL for TensorFlow/Pytorch) and HPC programming interfaces (e.g....Senior$152k - $241.5k
...and industries. Within our software stack, CUTLASS stands out as... ...(GEMM) and related math and deep learning computations on NVIDIA GPUs.... ...Computer Science, Computer Engineering, or related field (or equivalent... ...of DL frameworks like PyTorch, JAX, SGLang, vLLM, TRT-LLM,...Senior- We are now looking for a Senior Deep Learning Software Engineer, TensorRT Performance! NVIDIA is seeking an experienced Deep Learning Engineer passionate... .... Experience with deep learning frameworks (e.g., PyTorch, JAX, TensorFlow, ONNX) and inference libraries (e.g.,...Senior
$184k - $287.5k
We are looking for outstanding Senior Deep Learning Software Engineers to develop and productize NVIDIA's deep learning solutions in autonomous driving... ...workflows that let users leverage frameworks (e.g. PyTorch, JAX) and compiler technology tools (e.g. MLIR, Triton)...SeniorWork experience placement- Overview Reinforcement learning post‑training is... ...building an RL Frameworks engineering team to develop the open... ...team spans the full software stack, from collaborating... ...their need optimizing deep learning frameworks,... ...training and generation) PyTorch internals, including...Senior
$152k - $241.5k
Senior Software Engineer - Deep Learning Inference What you’ll be doing: Craft and develop robust inferencing software that can be scaled to multiple platforms... ...development Experience in working with TensorRT, PyTorch, TensorFlow, ONNX Runtime or other ML frameworks. Your...Senior$184k - $287.5k
Senior Software Engineer, AI Storage page is loaded## Senior Software Engineer, AI Storagelocations... ...parallel computing. More recently, GPU deep learning ignited modern deep learning — the... ...Databases* Internals of frameworks like PyTorch, JAX* Exceptional CUDA programming...Senior$152k - $241.5k
...with OS, container, GPU, and systems engineers, and apply machine learning or deep learning techniques for... ...classification or prediction) within existing software workflows. Qualifications 5+ years... ...Experience with TensorFlow or PyTorch in Linux and HPC or performance‑sensitive...Senior- ...in optimizing and developing deep learning frameworks for AMD GPUs.... ...collaborate across internal GPU software teams and engage with open‑... ...ecosystem. THE PERSON Skilled engineer with strong technical and... ...frameworks like TensorFlow, PyTorch, and SGLang on AMD GPUs via...Senior
$152k - $241.5k
## Senior Deep Learning Software Engineer - Autonomous VehiclesApplylocations: US, CA, Santa Clara: US, TX, Remote: US, VA, Remote: US, WA, Remotetime... ...in developing or using deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX, etc.)* Experience with solving...SeniorWork experience placementRemote work$184k - $287.5k
...for KV‑caching. Conduct deep, layer‑by‑layer model... .... Leverage the PyTorch ecosystem to extract standardized... ...road. Architect the software interface to... ...Computer Science, Computer Engineering, or a related technical... ...JAX, or similar machine learning frameworks....Senior$184k - $287.5k
Senior Software Engineer, CUDA Core Libraries page is loaded## Senior Software Engineer, CUDA Core... ...light experience for developers across deep learning, scientific computing, and data... ...experience with CUDA C++, CUDA Python, PyTorch, JAX, Numba, CuPy, or similar GPU-accelerated...SeniorFull time$184k - $287.5k
...looking for an experienced software professional to contribute to... ...NumPy, SciPy, TensorFlow and PyTorch. These frameworks provide an... ...computing, data analytics, deep learning, and professional graphics,... ...Science, Applied Math, Electrical Engineering or related field (or...Senior$184k - $287.5k
...for outstanding AI systems engineers to develop groundbreaking technologies... ...in the inference systems software stack! We build innovative... ...engineers at NVIDIA across deep learning frameworks, libraries,... ...deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX, etc)...Senior$184k - $287.5k
...seeking highly skilled and motivated software engineers to join us and build AI inference systems... ...programming, distributed systems, deep learning theories. Knowledgeable and passionate... ...performance engineering in ML frameworks (e.g., PyTorch) and inference engines (e.g., vLLM and...Senior$110k - $190k
We are hiring a Senior Software & AI Engineer to build production-grade AI systems, with a strong emphasis on deep learning and generative AI. You will work on specific components and... ...Experience with deep learning frameworks (PyTorch, TensorFlow, scikit‑learn)....Senior$224k - $356.5k
...Team is building the software foundation for... ...looking for exceptional engineers who thrive on... ...We are seeking a Senior Software Engineer... ...and deployment of deep neural networks that... ...core platform, deep learning inference,... ...TensorRT-LLM, ONNX, PyTorch, CUDA, Triton, or...SeniorFull time- ...revolution, building the software and systems that... ...We are looking for a Senior Software Engineer to lead the bring‑up,... ...scale. You will lead deep performance and reliability... ...intersection of deep learning systems, GPU... ...inference workloads using PyTorch, NeMo / Megatron,...Senior
$152k - $287.5k
...NVIDIA Gruppe is seeking a highly motivated Senior Software Engineer to join our communication libraries and network software team in Santa... ...computing systems used in High Performance Computing and Deep Learning. The ideal candidate should have a M.S./Ph.D. degree and...Senior- NVIDIA is looking for a Senior Deep Learning Software Engineer to enhance its TensorRT performance and inference ecosystem. The role focuses on optimizing performance across various NVIDIA accelerators and developing groundbreaking methodologies. Applicants should have...Senior
$152k - $241.5k
...for their CUTLASS team to benchmark and enhance the performance of deep learning models. A strong background in Python and C++ is essential, alongside advanced degrees and 3+ years of experience in software performance analysis. The role involves identifying performance...Senior$184k - $287.5k
NVIDIA is seeking a Senior Software Engineer, NCCL and CUDA specialization to join our Cloud Service... ...development. Good exposure to PCIe and NVLINK. Deep understanding of operating systems and... ...experience. Experience with deep learning workloads training and inferencing....Senior$184k - $287.5k
We are hiring software engineers to work on the CUDA driver, a core component of our platform for accelerating general purpose computation... ...a growing range of computational workloads, ranging from deep learning, scientific computation, and self-driving cars to video games...Senior- Overview Join the new Agentic Engineering team, within the Deep Learning Framework Group, at NVIDIA. We build the agentic workflows that automate code generation, testing, and tuning across NVIDIA's frameworks, compilers, and developer tooling. The team is a force multiplier...SeniorWork experience placement
$152k - $241.5k
NVIDIA Corporation is seeking a highly motivated Senior Software Engineer for its communication libraries and network software team in Santa... ...designing and maintaining communication runtimes for Deep Learning frameworks and participating in parallel programming specifications...Senior
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior Software Engineer, PyTorch - Deep Learning. Be the first to apply!
- software engineer amazon Santa Clara, CA
- agile software developer Santa Clara, CA
- rust software engineer Santa Clara, CA
- software developer positions Santa Clara, CA
- senior software design engineer Santa Clara, CA
- software developer Santa Clara, CA
- ngo software engineer Santa Clara, CA
- startup software engineer Santa Clara, CA
- software development engineer (robotics engineer) Santa Clara, CA
- scientific software engineer Santa Clara, CA

