Senior Machine Learning Engineer Job in Pune at NVIDIA

NVIDIA is hiring for the role of Senior Machine Learning Engineer!

Responsibilities of the Candidate:

• Designing, implementing, and optimizing generative AI models like Stable Diffusion, and Llama using Python, CUDA C/C++, and tools like ONNX and TensorRT.
• Creating reproducible deployments using Docker and implementing efficient model pipelines with Triton Inference Server.
• Collaborating with multi-functional teams to develop machine learning solutions.
• Acting as a technical liaison, guiding in customer-facing roles.
• Staying abreast of advancements in AI theory and generative AI models.

Requirements:

• 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.
• 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.