Senior Software Engineer, GNN

NVIDIA
Austin, Texas, California, United States
Workplace: RemoteFull timeUSD 152,000 - 287,500 annuallyFunction: Software EngineeringExperience: 5+ yearsEducation: bachelorsSkills: ["Collaboration","Communication","Technical leadership","Mentorship","Documentation"]

Build accelerated, PyTorch-based solutions for large-scale machine learning models, including graph neural networks, tabular foundation models, and ensemble methods. Work on GPU infrastructure to optimize training and inference workflows, integrating CUDA-X libraries into PyTorch pipelines. Partner with developers, product managers, and scientists to deliver customer-facing model development and prediction capabilities, while providing technical leadership and improving codebase architecture through re-architecture and agentic debugging tools.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
NVIDIA
NVIDIA
2 hours ago

Senior Software Engineer, GNN

✓ Verified Job

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

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

Job Summary

Build accelerated, PyTorch-based solutions for large-scale machine learning models, including graph neural networks, tabular foundation models, and ensemble methods. Work on GPU infrastructure to optimize training and inference workflows, integrating CUDA-X libraries into PyTorch pipelines. Partner with developers, product managers, and scientists to deliver customer-facing model development and prediction capabilities, while providing technical leadership and improving codebase architecture through re-architecture and agentic debugging tools.
Location: Austin, Texas, California, United States
Workplace: Remote
Employment Type: Full time
Job Function: Software Engineering
Seniority: Mid level

Key Responsibilities

  • •Develop accelerated, PyTorch-based solutions for large-scale ML models (GNNs, TFMs, ensemble models) with efficient GPU training and inference.
  • •Support CUDA-X libraries and integrations used in PyTorch-based, large-scale ML workflows.
  • •Partner with developers, product managers, and scientists to create GPU-accelerated GNN implementations for model development and prediction.
  • •Provide customer-facing support by gathering technical requirements and guiding product/engineering priorities.
  • •Provide technical leadership and mentorship, improve the codebase through re-architecture, and use agentic coding tools to debug, refactor, and implement features.

Pay and Benefits

Salary: USD 152,000 - 287,500 annually
Equity and Bonus:Equity

Key Requirements

  • •Bachelor’s degree (or equivalent) plus 5+ years in large-scale machine learning, deep learning, and data science; or a Master’s/PhD plus 3+ years.
  • •3+ years of experience with PyTorch.
  • •2+ years training enterprise-scale machine learning models across distributed infrastructure.
  • •2+ years designing and operating efficient GPU training/inference workflows (profiling, scaling, orchestration, resource utilization).
  • •Excellent C++ programming and software design skills, plus experience developing/debugging/optimizing high-performance GPU-accelerated applications (CUDA).
Experience:5+ yearsMachine learningDeep learningData scienceDistributed infrastructureGPU accelerationHigh-performance computing
Education:Bachelor's
Skills:CollaborationCommunicationTechnical leadershipMentorshipDocumentation
Tech Stack:PyTorchGraph Neural NetworksGNNTabular Foundation ModelsEnsemble modelsCUDA-XCUDAC++GPU accelerationPyTorch GeometricSnowflakeDatabricksFAISSMilvusVector searchLakehouseDistributed infrastructureProfilingOrchestration

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