Senior Tools Development Engineer – Machine Learning

Job Title

Senior Tools Development Engineer – Machine Learning

Role Summary

Join NVIDIA's Deep Learning Focus Group to design and implement deep learning and AI solutions for GPU validation and gameplay automation. The role focuses on building ML-driven test frameworks, intelligent gameplay agents, and end-to-end validation pipelines for video, image, and audio analysis on NVIDIA GPUs.

Experience Level

Senior — the posting requests at least 2 years of hands-on ML engineering experience.

Responsibilities

Key responsibilities include:

• Design and deploy intelligent gameplay agents using computer vision, reinforcement learning, imitation learning, and LLM/VLM-based agents.
• Apply ML/DL techniques for QA and defect detection across video, image, and audio pipelines.
• Develop and maintain Python-based automation pipelines that integrate neural networks for GPU validation.
• Establish scalable training and deployment environments using Linux, Docker, and TensorRT.
• Build, clean, augment, and manage datasets to support continuous, self-improving training pipelines.

Requirements

Must-have:

• 2+ years hands-on ML engineering experience.
• Proficiency in PyTorch and TensorFlow/Keras; experience with ONNX and TensorRT.
• Advanced Python (OOP, design) for large-scale applications.
• Practical experience with Linux and Docker.
• Experience with OpenCV and state-of-the-art DL algorithms for classification, detection, tracking, and segmentation.
• Familiarity with QA methodologies and NVIDIA GPU technologies (e.g., RTX, DLSS).
• Experience building agentic gameplay systems using LLM/VLM agents, GenAI, RAG, or vLLM and using AI developer tools for automation.
• Strong written and verbal communication and commitment to high software quality standards.

Nice-to-have:

• Experience with transformer-based LLMs, diffusion models, imitation learning, model-free/model-based RL, hierarchical or inverse RL, meta-learning, or lifelong learning.
• Practical research or competition experience in computer vision or data science (e.g., Kaggle).
• Demonstrated ability to decompose game scenarios into solvable DL problems and implement end-to-end solutions.

Education Requirements

Master's or PhD in AI, Machine Learning, Computer Science, or a related technical field (or equivalent practical experience) as specified in the posting.

About the Company

Company: NVIDIA

Headquarters: Santa Clara, California, USA

NVIDIA is a global leader in accelerated computing, renowned for its innovative solutions in AI and digital twins that transform diverse industries. The company specializes in networking technologies, providing end-to-end InfiniBand and Ethernet solutions for servers and storage that optimize performance and scalability. NVIDIA serves sectors such as high-performance computing, enterprise data centers, and cloud computing, constantly reinventing its products and services to stay ahead in the market.