Senior Tools Development Engineer – Machine Learning

Overview

• This role within Deep Learning Focus Group is strongly technical, responsible for building DL/AI based solutions for validation of NVIDIA GPUs.
• GPU Render Output Analysis (Video, Images, Audio) and complex problems like Intelligent Game Play Automation.
• This person would need to analyze/understand the challenges from stakeholders of various groups, design & implement DL solutions to resolve them.
• We are specifically looking for expertise in the following key areas: Game play automation using deep learning (DL) and artificial intelligence (AI) techniques, with a focus on research and application in the gaming industry.The candidate should be proficient in the following DL techniques: Deep Reinforcement Learning (DRL) and Imitation Learning Vision Transformers (ViTs) and Convolutional Neural Networks (CNNs) Generative AI and Diffusion Models Prior experience or research in AI-driven bots, In-Game Movement Automation, and Player Behavior Prediction are highly desirable.
• Knowledge in using AI development tools for test plans creation, test cases development and test cases automation.
• What you'll be doing: Build Intelligent Gameplay Automation & Agentic Workflows: Design and deploy advanced gameplay agents using computer vision, reinforcement learning, imitation learning, and LLM/VLM-based agents, leveraging state-of-the-art tools like Codex and Claude.
• Drive ML-Driven QA & Defect Detection: Apply ML/DL techniques to solve complex QA challenges across NVIDIA product lines, implementing DL-based solutions for video/audio defect detection and optimizing automated test frameworks to boost productivity.
• Develop End-to-End GPU Validation Solutions: Create and maintain robust, Python-based automation pipelines that consume neural networks to rigorously validate NVIDIA GPUs.
• Establish Scalable Infrastructure & Deployment: Set up and manage scalable development environments using Linux, Docker, and TensorRT to train, validate, and deploy large-scale neural networks.
• Curate Self-Improving Data Pipelines: Build, clean, and augment high-quality datasets to feed and enable continuous, self-improving training pipelines.
• What we need to see: Master’s or PhD in AI/ML/CS (or equivalent) with at least 2 years of hands‑on ML engineering experience.
• Extensive knowledge of PyTorch, TensorFlow/Keras, ONNX, and TensorRT.
• Advanced Python proficiency with strong OOP, design, and problem-solving skills for large-scale applications, combined with familiarity and hands-on experience in Linux and Docker.
• Solid understanding of OpenCV and state-of-the-art DL algorithms for image classification, object detection, tracking, and segmentation.
• Well-versed in QA methodologies with a deep understanding of NVIDIA GPU technologies (e.g., RTX, DLSS).
• Hands-on experience building agentic gameplay systems using LLM/VLM-based agents, GenAI, RAG, vLLM, and solving complex problems with AIGC; proficient with AI development tools (Codex, Cursor, MCP, CodeRabbit) for test automation and workflow acceleration.
• Excellent written and verbal communication, strong initiative, self-motivation, and a commitment to high software quality standards.
• Ways to stand out from the crowd: Knowledge of Transformer based LLM and AIGC, Imitation Learning, Model free/based RL, Hierarchical RL, Inverse RL, Meta-learning, Life-long learning.
• Hands-on experience in solving complex problems using Deep learning Algorithms would be a plus.
• Experience and medals in data science or computer vision competitions (e.g., Kaggle, CVPR workshop) will be a plus.
• Demonstrated ability to rapidly understand game dynamics and decompose game scenarios into solvable DL problems that can be implemented end‑to‑end.