Senior Deep Learning Framework Communications Engineer
$152k - $241.5kSwiftCruit
- Senior Deep Learning Framework Communications EngineerOnsite |Senior Level|Full TimePosted: Yesterday#### Company LocationUS, CA, Santa Clara, United States of America#### Remote Work PolicyOnsite#### SkillsPythonC++Fault ToleranceSOLIDOperating SystemsCompiler DesignPyTorchvLLMDeep LearningLLMNVIDIA is leading the way in groundbreaking developments in Artificial Intelligence, High Performance Computing and Visualization. The GPU, our invention, serves as the visual cortex of modern computers and is at the heart of our products and services. Our work opens up new universes to explore, enables amazing creativity and discovery, and powers what were once science fiction inventions from artificial intelligence to autonomous cars.We are looking for a motivated Deep Learning engineer to bring advanced communication technologies into AI stacks, including PyTorch, TRT-LLM, vLLM, SGLang, JAX, etc. You will be working with the team that created communication libraries like NCCL, NVSHMEM & technology like GPUDirect -- for scaling Deep Learning and HPC applications. Your customers will have diverse multi-GPU demands, ranging from training on scales up to 100K GPUs to inference down at microsecond latency. Communication performance between the GPUs has a direct impact on AI applications. Your work in AI toolkits will make all of those easier for the community. This is an outstanding opportunity for someone with an AI background to advance the state of the art in this space. Are you ready to contribute to the development of innovative technologies and help realize NVIDIA's vision?**What you will be doing:*** Integrate new communication libraries features in AI frameworks: from PoC to performance analysis to production* Perform deep analysis of AI workloads and frameworks to identify multi-GPU communication requirements and opportunities. Collaborate hands-on with teams working on the latest AI models.* Improve AI compilers to hide communications or perform automatic fusion.* Conduct in-depth AI workload performance characterization on multi-GPU clusters.* Design fault-tolerant and elastic solutions for large-scale or dynamic AI workloads.* Author custom communication or fused compute-communication kernels to showcase ultimate performance on NV platforms.* Influence the roadmap of communication libraries - NCCL & NVSHMEM.* Collaborate with a very dynamic team across multiple time zones.**What we need to see:*** B.S, M.S. or PHD in Computer Science, or related field (or equivalent experience) with 5+ software engineering and HPC/AI experience* Development or integration experience with Deep Learning Frameworks such PyTorch, JAX, and Inference Engines such as TRT-LLM, vLLM, SGLang* Rapid prototyping and development with Python, C++, CUDA or related DSLs (Triton, cuTe)* Solid grasp of AI models, parallelisms, and/or compiler technologies (e.g. torch.compile)* Experience conducting performance benchmarking on AI clusters. Familiarity with at least one performance profiler toolchain (PyTorch profiler, NVIDIA Nsight Systems)* Understanding of HPC/AI communication concepts (1-sided v 2-sided communication, elasticity, resiliency, topology discovery, etc)* Adaptability and passion to learn new areas and tools* Flexibility to work and communicate effectively across different teams and timezones**Ways to stand out from the crowd:*** Experience with parallel programming on at least one communication runtime (NCCL, NVSHMEM, MPI). Good understanding of computer system architecture, HW-SW interactions and operating systems principles (aka systems software fundamentals)* Expertise in one or more of these areas: Training, Distributed inference, MoE, Reinforcement Learning, kernel authoring (on CUDA, Triton, cuTe, etc). Experience with programming for compute & communication overlap in distributed runtimes* Experience with AI compiler pattern matching and lowering. Solid understanding of memory hierarchy, consistency model, and tensor layoutYour base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 152,000 USD - 241,500 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4.You will also be eligible for equity and benefits.Applications for this job will be accepted at least until July 6, 2026.This posting is for an existing vacancy.NVIDIA uses AI tools in its recruiting processes.NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) 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 SwiftCruit
Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the Senior Deep Learning Framework Communications Engineer in Santa Clara, CA vacancy
$152k - $287.5k
NVIDIA Gruppe is hiring software engineers for its Deep Learning Frameworks Sustaining Engineering team in Santa Clara, California. In this role, you'll back-port changes and track dependencies while contributing to TensorFlow, PyTorch, and TensorRT products. The ideal...Senior- SwiftCruit is seeking a Senior Deep Learning Framework Communications Engineer in Santa Clara, California. This role involves integrating communication libraries with AI frameworks like PyTorch and performing deep analysis of multi-GPU demands. Ideal candidates will have...Senior
- Advanced Micro Devices in Santa Clara is seeking an Industry Leading Expert C++ Developer to enhance deep learning frameworks for AMD GPUs. You will develop and optimize GPU kernels while collaborating with internal and open-source teams, utilizing cutting-edge compiler...Senior
$168k - $258.75k
## Senior Product Manager, AI FrameworksApplylocations... ...build best in class frameworks in OSS and support a... ...We focus on enabling deep learning across all GPU use... ...Computer Science, Computer Engineering, or similar... ...technology company* Strong communication and interpersonal...Senior$152k - $218.5k
NVIDIA is hiring software engineers for its Deep Learning Frameworks Sustaining Engineering team. Our team produces software that's powering a revolution... ...to the stability of releases, and effectively communicate status to those involved in the release in a detail-oriented...Senior- Overview Reinforcement learning post‑training is driving some of... .... NVIDIA is building an RL Frameworks engineering team to develop the open‑... ...address their need optimizing deep learning frameworks, or... ...Strong verbal and written communication skills and the ability to collaborate...Senior
$240k - $320k
...experts for product engineering of AI-based... ...Description As the Senior Principal... ...E2E AI Training Framework for Autonomous Driving... ...deployments ~ Deep knowledge of E2E... ...including reinforcement learning, and imitation... ...Exceptional leadership and communication skills, with the...SeniorFull timeWork experience placementLocal areaFlexible hoursBosch Group
Sunnyvale, CA4 days ago$272k - $431.25k
..., we are seeking exceptional engineers to join our autonomous driving... ...imitation, and reinforcement learning—to improve the planning and... ...passionate about autonomous systems. Deep understanding of modern deep... ...with major deep learning frameworks. Familiarity with C++ for...SeniorWork experience placement- Dormont Manufacturing Co is seeking a Senior Level 4 Support Engineer in Sunnyvale, California. This role involves bridging customer environments with internal development, troubleshooting complex issues, and providing product feedback. The ideal candidate will have over...Senior
- NVIDIA Corporation is seeking a Senior Product Manager for AI Frameworks based in Santa Clara, California. In this role,... ...management, a solid grounding in machine learning and distributed systems, as well as excellent communication skills. Attractive salary and benefits offered...Senior
$129k - $198.4k
General Motors is seeking an AI/ML Engineer for the Metrics Frameworks team in Sunnyvale, California. The successful candidate will focus on developing analytics frameworks and tools to accelerate autonomous vehicle development and testing. Candidates should have a BS in...Senior- ...developing and maintaining production-grade iOS applications, with strong knowledge in SwiftUI and UIKit. Familiarity with Apple frameworks is essential for this role. Join a dynamic team that values inclusive talent and innovation. #J-18808-Ljbffr Quantum Technologies...Senior
- NVIDIA Gruppe is seeking a Senior Deep Learning Software Engineer to develop cutting-edge deep learning solutions in autonomous driving vehicles. You... ...significant experience in software development and deep learning frameworks. Join us in advancing innovations in Physical AI. #J-1...Senior
- NVIDIA Corporation is seeking a Senior Deep Learning Software Engineer to analyze and improve the performance of NVIDIA's inference ecosystem. This role requires significant expertise in deep learning frameworks and performance optimization. The ideal candidate will have...Senior
- ...iPhone and iPad, utilizing the latest programming features. Ideal candidates possess 8+ years of hands-on experience and a deep understanding of Apple frameworks. The role offers competitive compensation in a diverse workplace. #J-18808-Ljbffr Robotics Technologies LLCSenior
$184k - $356.5k
NVIDIA Corporation is looking for a Senior Engineer to own and evolve the core NIM Platform SDK and microservice framework. This role involves developing high-performance systems programming and collaborating across teams to deliver AI inference at scale. The ideal candidate...Senior- A leading technology company is seeking a Senior AI Software Engineer to join their team in Santa Clara, California. In this role, you will... ...systems software for inference applications including deep learning framework optimizations and GPU kernel technologies. You will...Senior
$224k - $356.5k
NVIDIA Gruppe is seeking a Senior Deep Learning Software Engineer in Santa Clara to design and build an automated inference and deployment solution.... ...defining a scalable DL architecture that integrates with frameworks like PyTorch and JAX. Ideal candidates should have over...Senior- A leading technology company is seeking a skilled engineer to optimize deep learning frameworks and enhance GPU kernel performance. The ideal candidate excels in collaborative and independent work, with a strong background in C++ and Python. Responsibilities include optimizing...Senior
- ...candidate should have over 8 years of experience developing production-grade iOS applications, as well as a solid understanding of Apple frameworks. This role emphasizes modern UI development and requires expertise in memory and battery management. Benefit from working in an...Senior
$224k - $356.5k
NVIDIA Gruppe, located in Santa Clara, CA, is seeking a Senior Deep Learning Systems Engineer to advance its Autonomous Vehicles project. In this... ...in the field and a strong background in deep learning frameworks, particularly PyTorch. NVIDIA offers a competitive salary...Senior- A leading technology firm based in Sunnyvale, California is looking for a Senior Systems Framework Engineer to architect and implement frameworks for innovative products. You will focus on cross-device capabilities and work on distributed systems and multimedia processing...Senior
- ...NVIDIA Corporation is seeking a Senior Software Architect to enhance communication libraries crucial for scaling Deep Learning and HPC applications. You will investigate performance... ...C/C++, parallel programming, and AI/HPC frameworks. This position is based in Santa Clara,...SeniorRemote job
- ...is seeking a skilled professional to develop leading machine learning frameworks, focusing on weather and climate AI. The role involves collaborating... ...) and possess strong Python skills along with experience in deep learning frameworks. A competitive salary range is offered,...Senior
- ...The ideal candidate will have extensive experience in machine learning, deep learning, and developing domain-specific models. This role... ...Ph.D. in a related field and possess strong programming skills with Python and relevant frameworks. #J-18808-Ljbffr Applied MaterialsSenior
$152k - $241.5k
...Gruppe in Santa Clara is seeking a Sr. Software Engineer specializing in AI-driven semiconductor... ...Applicants should have a strong background in deep learning and computer vision, with proficiency in Python and modern frameworks like PyTorch or TensorFlow. The salary range...Senior- NVIDIA Corporation is searching for a Senior Deep Learning Software Engineer in Santa Clara, California. In... ...contribute to the PyTorch deep learning framework. Applicants should have a BS in... ..., proficient in C++, and strong communication skills. The position offers...Senior
$152k - $287.5k
NVIDIA Gruppe is seeking a Senior Machine Learning Applications and Compiler Engineer in Santa Clara, California. This role involves developing algorithms... ...engineering skills, and familiarity with deep learning frameworks. The position offers a competitive salary range...Senior$184k - $356.5k
NVIDIA Gruppe is seeking a Senior Engineer to lead the evolution of the core NIM Platform SDK and microservice framework in Santa Clara, California. This hands-on role involves architecting significant new features and solving complex engineering challenges. The ideal candidate...Senior$224k - $356.5k
NVIDIA in Santa Clara is seeking a Senior Deep Learning Systems Engineer to advance its Autonomous Vehicles project. This role will focus on building... ...system architecture, strong knowledge of deep learning frameworks, and proficiency in Python and C++. A competitive...Senior
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior Deep Learning Framework Communications Engineer. Be the first to apply!
Related searches
- digital communications systems engineer Santa Clara, CA
- senior game producer Santa Clara, CA
- senior electronic design engineer Santa Clara, CA
- senior robotics software engineer Santa Clara, CA
- senior specialist Santa Clara, CA
- senior software test engineer Santa Clara, CA
- sr project engineer Santa Clara, CA
- senior electrical designer Santa Clara, CA
- senior director clinical development Santa Clara, CA
- senior logistics manager Santa Clara, CA