Posted Aug 16

Nvidia is hiring a
Senior Machine Learning Engineer

India, Pune • 2 Locations
Full time

We are looking for a Senior Machine Learning Engineer with expertise in artificial intelligence, data science, and machine learning. You should be technically skilled, solution-oriented, with keen interest in AI theory and generative AI models! NVIDIA’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world! Today, we are increasingly known as “the AI computing company.” Are you willing to challenge yourself, and build phenomenal software by working with the smartest people in the world? Join us at the forefront of technological advancement.

We are accelerating enterprise AI efforts and creating innovative solutions by harnessing the power of machine/deep learning techniques to address the challenges posed by high-volume and high-velocity data. The team is looking for candidates with a strong understanding of machine learning, particularly focusing on times-series, NLP, deep learning (e.g. transformers), and reinforcement learning. Candidates should ideally have hands-on experience working with deep learning frameworks such as PyTorch, TensorRT, Keras, or similar tools. Moreover, it would be beneficial if candidates have experience in designing and deploying complex ML pipelines in production environments.

What you’ll be doing:

  • Design, implement, and optimize generative AI models like Stable Diffusion, Llama using Python, CUDA C/C++, and tools like ONNX and TensorRT.

  • Create reproducible deployments using Docker and implement efficient model pipelines with Triton Inference Server.

  • Collaborate with multi-functional teams to develop machine learning solutions.

  • Act as a technical liaison, guiding in customer-facing roles.

  • Stay abreast of advancements in AI theory and generative AI models.

What we need to see:

  • 3+ years in artificial intelligence data science, machine learning, or related fields.

  • Proficiency in Python, CUDA C/C++, and machine learning libraries like PyTorch and TensorRT.

  • Experience with Text-to-Image models such as Stable Diffusion.

  • Experience with Docker containers and Triton Inference Server.

  • Knowledge of AI theory and generative AI models like stable diffusion, LLMs, langchain, and llamaindex.

  • Previous technical sales engineer experience in a cloud computing environment or equivalent.

  • Exceptional problem-solving skills, communication skills, and ability to work in multi-functional teams.

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people on the planet working for us. If you're creative and autonomous, we want to hear from you! You will also be eligible for equity and benefits.

NVIDIA is committed to fostering a diverse 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.

Please mention that you found the job on ARVR OK. Thanks.