Senior Deep Learning Software Engineer, Inference and Model Optimization

NVIDIA is seeking a Senior Deep Learning Software Engineer to develop and scale up our automated inference and deployment solution. The team focuses on optimizing generative AI models for maximal inference efficiency using techniques such as neural architecture search and pruning. The ideal candidate will have a strong background in machine learning fundamentals and software architecture & engineering, and will be instrumental in pushing the limits of inference efficiency and large-scale, automated deployment.

Requirements

• Masters, PhD, or equivalent experience in Computer Science, AI, Applied Math, or related field.
• 5+ years of relevant work or research experience in Deep Learning.
• Excellent software design skills, including debugging, performance analysis, and test design.
• Strong proficiency in Python, PyTorch, and related ML tools (e.g. HuggingFace).
• Strong algorithms and programming fundamentals.
• Good written and verbal communication skills and the ability to work independently and collaboratively in a fast-paced environment.

Benefits

• Generous Paid Time Off
• 401k Matching
• Retirement Plan
• Visa Sponsorship