Senior Deep Learning Engineer - AI for Wireless Systems
NVIDIA
Job Summary
NVIDIA is seeking a Senior Deep Learning Engineer to join its global 6G initiative at the Vietnam R&D Center. This role involves designing and prototyping deep learning models for AI-native wireless networks, focusing on signal processing tasks like channel estimation and beam alignment. The engineer will implement, train, and validate neural networks using PyTorch or TensorFlow, collaborate on integrating models into full-stack RAN, and optimize performance for real-time inference on GPU platforms. This position is crucial for shaping the future of AI, wireless communications, and high-performance computing.
Must Have
- Design and prototype deep learning models for wireless signal processing tasks.
- Work with simulation tools and real-world datasets to build models.
- Implement, train, and validate neural networks (e.g., CNNs, Transformers, GNNs) using PyTorch or TensorFlow.
- Collaborate with researchers and system engineers to integrate models into full-stack RAN.
- Optimize model performance for real-time inference and hardware acceleration.
- Contribute to model evaluation, benchmarking, and deployment readiness on GPU platforms.
- MS or PhD in Electrical Engineering, Computer Engineering, or a related field (or equivalent experience).
- 12+ years of experience in wireless communications, signal processing, or AI/ML.
- Deep understanding of communication systems (e.g., MIMO, OFDM, fading channels) and DSP fundamentals.
- Strong experience in training and deploying deep learning models for time-series or signal-based tasks.
- Proficiency in Python and experience with DL frameworks like PyTorch or TensorFlow.
- Familiarity with tools such as MATLAB, GNU Radio, or NVIDIA Sionna for wireless simulation.
Good to Have
- Experience with AI for 5G/6G systems, AI-for-RAN architecture, or telecom-grade deployments.
- Knowledge of channel estimation by AI, model compression, real-time inference, or GPU optimization.
- Exposure to CUDA, Triton, or real-time inference pipelines.
- Contributions to research publications or open-source wireless/AI projects.
Job Description
Job Requisition ID
JR2007909
Job Category
Engineering
Time Type
Full time
NVIDIA is looking for exceptional technical talent to help build the future of AI-native wireless networks. As part of our global 6G initiative, we are developing next-generation signal processing and radio access technologies that integrate deep learning at their core. From physical layer optimization to full-stack AI-powered RAN, we’re reimagining how wireless systems are designed, trained, and deployed. At NVIDIA Vietnam R&D Center (VRDC), you’ll join a world-class team shaping the next wave of innovation at the intersection of deep learning, wireless communications, and high-performance computing.
What You’ll Be Doing:
- Design and prototype deep learning models for wireless signal processing tasks such as channel estimation, beam alignment, link adaptation, and scheduling.
- Work with simulation tools and real-world datasets to build models that generalize across diverse wireless scenarios.
- Implement, train, and validate neural networks (e.g., CNNs, Transformers, GNNs) using PyTorch or TensorFlow.
- Collaborate with researchers and system engineers to integrate models into full-stack RAN.
- Optimize model performance for real-time inference and hardware acceleration.
- Contribute to model evaluation, benchmarking, and deployment readiness on GPU platforms.
What We Need to See:
- MS or PhD in Electrical Engineering, Computer Engineering, or a related field (or equivalent experience).
- 12+ years of experience in wireless communications, signal processing, or AI/ML.
- Deep understanding of communication systems (e.g., MIMO, OFDM, fading channels) and DSP fundamentals.
- Strong experience in training and deploying deep learning models for time-series or signal-based tasks.
- Proficiency in Python and experience with DL frameworks like PyTorch or TensorFlow.
- Familiarity with tools such as MATLAB, GNU Radio, or NVIDIA Sionna for wireless simulation.
Ways to Stand Out from the Crowd:
- Experience with AI for 5G/6G systems, AI-for-RAN architecture, or telecom-grade deployments.
- Knowledge of channel estimation by AI, model compression, real-time inference, or GPU optimization
- Exposure to CUDA, Triton, or real-time inference pipelines.
- Contributions to research publications or open-source wireless/AI projects.
At NVIDIA, you will join a world-class team building breakthrough technologies at the intersection of AI and wireless systems. Our Vietnam R&D Center is rapidly growing as a hub of innovation and collaboration, shaping the future of AI-native infrastructure. If you are passionate about AI, excited by wireless systems, and ready to help define the future of 6G, we want to hear from you.