Senior AI Researcher - BioNemo and TAO

NVIDIA
Vietnam
Workplace: OnsiteFull timeFunction: Research & Scientific (R&D)Experience: 5+ yearsSkills: ["Analytical skills","Problem-solving","Communication","Collaboration"]

Develop and optimize deep learning and generative AI models for scientific workloads, advancing NVIDIA’s BioNemo/TAO platform. You’ll design diffusion, graph neural network, and transformer-based architectures for structured prediction, build generative systems for molecular and abnormal image/video generation, and scale foundation models for domain applications. Partner with cross-functional teams to integrate research into SDKs and production pipelines using NIM and GPU acceleration techniques.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
NVIDIA
NVIDIA
4 weeks ago

Senior AI Researcher - BioNemo and TAO

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 8 hours agoStatus: Live

Job Summary

Develop and optimize deep learning and generative AI models for scientific workloads, advancing NVIDIA’s BioNemo/TAO platform. You’ll design diffusion, graph neural network, and transformer-based architectures for structured prediction, build generative systems for molecular and abnormal image/video generation, and scale foundation models for domain applications. Partner with cross-functional teams to integrate research into SDKs and production pipelines using NIM and GPU acceleration techniques.
Location: Vietnam
Workplace: Onsite
Employment Type: Full time
Job Function: Research & Scientific (R&D)
Seniority: Mid level

Key Responsibilities

  • •Investigate, design, and optimize deep learning models (diffusion models, flow-matching networks, transformer architectures, graph neural networks) for structured prediction in molecular and protein science.
  • •Build and implement generative AI systems for user cases such as molecular building, structural prediction, and image/video abnormal generation with reliable ranked outputs.
  • •Contribute to the development and scaling of foundation models for domain applications including protein structure and molecular interaction.
  • •Apply NVIDIA acceleration techniques and deploy large-scale AI models on NVIDIA GPU hardware with high efficiency and accuracy.
  • •Integrate research into BioNemo SDK and production workflows (including NIM), and build evaluation/post-processing pipelines; participate in reviews, documentation, and research discussions.

Key Requirements

  • •Bachelor’s (Honours), MS, or PhD (or equivalent experience) in Computer Science, Computer Engineering, Electrical Engineering, or a closely related field.
  • •5+ years of experience in machine learning and deep learning with hands-on experience in PyTorch (or equivalent frameworks).
  • •Strong grasp of generative models (diffusion models, flow matching, VAEs), transformer architectures, or graph neural networks for structured/geometry data.
  • •Proficiency in Python and experience running reproducible ML experiments (e.g., with Hydra, WandB, or similar tooling).
  • •Excellent analytical and problem-solving skills with the ability to implement and iterate complex model architectures.
Experience:5+ yearsMachine learningDeep learningGenerative AIDrug discoveryLife sciences
Education:
Skills:Analytical skillsProblem-solvingCommunicationCollaboration
Tech Stack:PyTorchPythonHydraWandBDiffusion modelsFlow-matching networksTransformer architecturesGraph neural networksGenerative AIVAEsCuEquivarianceEfficient attentionMixed-precision inferenceNVIDIA SDKsBioNemoTAOInference Microservices (NIM)GPU hardwareAblation studiesBenchmarking

Company Brief

NVIDIA
Designs and manufactures GPUs, AI accelerators, and system-on-chip products for gaming, data centers, professional visualization, and automotive markets, enabling advanced graphics, AI, and high-performance computing solutions worldwide.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1993
Glassdoor
Glassdoor: 4.3
WebsiteLinkedInGlassdoor