Senior Tools Development Engineer – Machine Learning

NVIDIA NVIDIA · Semiconductors · Shanghai, China

Senior Tools Development Engineer at NVIDIA focused on building AI/ML solutions for validating NVIDIA GPUs, particularly in game rendering analysis and intelligent game play automation. The role involves designing and implementing deep learning techniques like DRL, ViTs, CNNs, and Generative AI, with a focus on agentic workflows, ML-driven QA, and end-to-end validation pipelines.

What you'd actually do

  1. 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.
  2. 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.
  3. Develop End-to-End GPU Validation Solutions: Create and maintain robust, Python-based automation pipelines that consume neural networks to rigorously validate NVIDIA GPUs.
  4. 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.
  5. Curate Self-Improving Data Pipelines: Build, clean, and augment high-quality datasets to feed and enable continuous, self-improving training pipelines.

Skills

Required

  • PyTorch
  • TensorFlow/Keras
  • ONNX
  • TensorRT
  • Python
  • Linux
  • Docker
  • OpenCV
  • image classification
  • object detection
  • tracking
  • segmentation
  • agentic gameplay systems
  • LLM/VLM-based agents
  • GenAI
  • RAG
  • vLLM
  • AIGC
  • Codex
  • Cursor
  • MCP
  • CodeRabbit

Nice to have

  • Deep Reinforcement Learning (DRL)
  • Imitation Learning
  • Vision Transformers (ViTs)
  • Convolutional Neural Networks (CNNs)
  • Generative AI
  • Diffusion Models
  • Transformer based LLM
  • Model free/based RL
  • Hierarchical RL
  • Inverse RL
  • Meta-learning
  • Life-long learning
  • data science competitions
  • computer vision competitions

What the JD emphasized

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

Other signals

  • building AI-based solutions
  • gameplay automation using deep learning
  • ML-driven QA
  • GPU validation solutions