All vacancies
Senior System Software Architect, HPC and AI Networking
NVIDIA
China, Shanghai · On-site · relocationSalary not disclosedfull-timeNVIDIA Careers (offices abroad)
NVIDIA has been redefining computer graphics, PC gaming, and accelerated computing for more than 25 years. Today, we lead in artificial intelligence, driving advances in natural language processing, computer vision, autonomous systems, and scientific research.
Responsibilities
- Design and prototype scalable software systems that optimize distributed AI training and inference—focusing on throughput, latency, and memory efficiency.
- Develop and evaluate enhancements to communication libraries such as NCCL, UCX, and UCC, tailored to the unique demands of deep learning workloads.
- Collaborate with AI framework teams (e.g., TensorFlow, PyTorch, JAX) to improve integration, performance, and reliability of communication backends.
- Contribute to the evolution of runtime systems, communication libraries, and AI-specific protocol layers.
- Collaborate with customers to understand their needs and provide innovative solutions for them.
Languages
- Work format
- On-site
- Seniority
- Senior
- Posted
- 31 Aug 2026
- Last verified
- In the last 5 days
Similar roles
- Solution ArchitectExelab S.r.l. · Remote · worldwide, UTC+4 accepted
- Senior Solutions Architect | Python and GenAIProvectus · Remote · Armenia
- .NET Solution ArchitectEPAM Systems · Armenia · Hybrid
- Solution Architect SAP DM (m/w/d)cbs Corporate Business Solutions · Remote · worldwide, UTC+4 accepted
- Banco de Talentos: Data Platform EngineerArco Educação · Remote · worldwide, UTC+4 accepted
- Performance Engineer - Performance Analysis & TuningSupabase · Remote · worldwide