Senior Software Engineer, GNN

NVIDIA is a technology company developing advanced computing and AI solutions. The company is seeking a Senior Software Engineer to build and optimize PyTorch-based large-scale machine learning solutions, including Graph Neural Networks, Tabular Foundation Models, and ensemble models, on GPU infrastructure. The role involves software development, customer collaboration, technical leadership, architecture improvements, and cross-team coordination.

Responsibilities:

• Develop accelerated, PyTorch-based solutions for large-scale machine learning models, including GNNs, TFMs, and ensemble models, with a focus on efficient training and inference on GPU infrastructure
• Support CUDA-X Libraries and integrations used in PyTorch-based, large-scale machine learning workflows
• Partner with developers, product managers, and scientists to develop innovative GNN models and GPU-accelerated implementations for model development and prediction phases
• Develop solutions that help customers adopt NVIDIA hardware and software, and gather technical requirements directly from customers and Solutions Architects to guide product and engineering priorities
• Provide technical leadership and mentorship to engineers across the team
• Identify opportunities to improve the codebase and reduce code-maintenance overhead through re-architecture
• Apply agentic coding tools to identify and fix bugs, implement new features, and refactor code
• Solve complex technical issues, explain solutions clearly, exercise technical leadership, and coordinate across multiple teams to achieve shared objectives