Associate AI/ML Engineer - Generative AI & Deep Learning
Bengaluru / Pune, IndiaPosted 11/2/2026Deadline: 12/12/2026
Work Mode: HYBRID
Employment: FULL TIME
Experience: 0 - 1 years of experience or research in Machine Learning / Deep Learning
₹18,00,000 - ₹26,00,000 / year + Equity
Openings: 8
Veyrai AI Preparation Recommendation
Generate a role-specific roadmap and practice voice technical interviews calibrated to NVIDIA.
Original Source: NVIDIA University Recruiting
About the Role
Join NVIDIA's AI platform team to accelerate large language models, inference pipelines, and neural network architectures on CUDA and TensorRT.
Key Responsibilities
- Optimize deep learning models for low-latency inference using TensorRT and Triton Inference Server.
- Benchmark training throughput and memory consumption across multi-GPU clusters.
- Implement custom CUDA kernels or PyTorch extensions for novel transformer layers.
- Collaborate with AI researchers to deploy production vision and language models.
Eligibility & Required Qualifications
Education:B.Tech/M.Tech/MS in Computer Science, AI, Robotics, or Electrical Engineering
Target Graduation Batches:2024 / 2025
- Strong Python and C++ programming skills.
- Deep theoretical and practical knowledge of neural networks, transformers, and CNNs.
- Proficiency in PyTorch, JAX, or TensorFlow.
- Understanding of linear algebra, probability, and optimization algorithms.
Required Technical Skills & Tools
PythonPyTorchC++CUDADeep LearningTransformersTensorRTLinear AlgebraPyTorchCUDADockerTritonLinuxGit
Preferred Qualifications
- Hands-on experience with CUDA C/C++ or GPU profiling (Nsight Systems).
- Published research or open-source Hugging Face model contributions.
Benefits & Perks
NVIDIA Employee Stock Purchase Plan (ESPP)
Generous Health Coverage
Hybrid Work Flexibility
Continuous Learning Budget
NVIDIA
Semiconductors & AI HardwareNVIDIA is the pioneer of GPU-accelerated computing and full-stack generative AI computing platforms.
Headquarters:Santa Clara, CA & Bengaluru/Pune, India
Company Size:25,000+ employees
Interview Readiness
Practice for NVIDIA with Sub-200ms Voice AI
Experience the actual verbal pressure of high-bar engineering interviews. Veyrai listens to your code explanations and interrogates system edge cases.