Nvidia Senior Systems Software Engineer TAO Machine Learning Data Modeling 

United States, California 

549867241

What you’ll be doing:

• Help in finding and creating (synthetic generation using GenAI/Simulation) the right data for a Multi-Modal model with scalable systems.

• Design various (ML and DL) architectures and loss functions to ingeniously formulate automated pseudo-labeling and GenAI for various multi-modal tasks.

• Design and develop an active (and passive) learning paradigm within (and out) of the loop annotators to iteratively mine informative data.

• Design insightful metrics (in settings: unsupervised, semi-and-supervised) for performance characterization of various models and data.

• Build scalable and robust ETL pipelines using novel and meaningful ML and DL models to deliver high-quality datasets.

• Work with internal teams to define requirements, enhance products, and automate workflows.

What we need to see:

• Bachelor's degree in Computer Engineering, Computer Science, Electrical Engineering, Robotics, or related field (or equivalent experience).

• 5+ years of ML / DL-related engineering experience with strong architecture and design skills.

• Excellent background and understanding of the deep roots of ML and DL.

• Proficient in understanding of perception systems, 2D or 3D and/or Temporal.

• Expertise with an understanding of out-of-distribution and related concepts.

• Knowledge of PyTorch, distributed machine learning, and distributed file systems.

• 5+ years leading complex sometimes ambiguous projects, particularly in high-throughput services at supercomputing scale.

Ways to stand out from the crowd:

• Good familiarity with multiple perception domains - Object detection, Segmentation, Multiple Object Tracking, Metric Learning.

• Knowledge of internal workings of Diffusion models.

• Familiarity with 3D geometrical aspects of Simulation and Inverse Computer Graphics.

• Proficient in running applications on cloud platforms using Kubernetes and Docker, and ML frameworks like Pytorch.

• Proficient in building systems and familiar with deep learning architectures and tools like NVIDIA TensorRT-LLM, Multimodal-LLM, and Triton Server.