Senior Software Engineer, GNN

Job not on LinkedIn

🔥 0 minutes ago

🏄 California, Texas – Remote

🏄 California – Remote 🤠 Texas – Remote

💵 $152k - $287.5k / year

⏰ Full Time

🟠 Senior

🧑‍💻 Full-stack Engineer

🦅 H1B Visa Sponsor

This company has sponsored H1B visas in the past.

👻 Ghost score 1%

What it means: how likely this posting is sitting open without active hiring.How to read it: 60% or higher means we think this is a ghost. Lower scores are less of a worry.How we score it: age, reposts, open roles vs team size, how fast jobs close, and whether pay is listed.Fine print: an estimate from public data, not a judgment of the company. Real jobs can score high, so treat it as a hint, not a fact.

Apply Now

Find Similar Remote Jobs

Set Up Email Alert For New Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

NVIDIA

WebsiteLinkedInAll Job Openings

10,000+ employees

Founded 1993

🏥 Healthcare

🏭 Manufacturing

🤖 Artificial Intelligence

Healthcare • Manufacturing • Artificial Intelligence

NVIDIA is a leading technology company specializing in accelerated computing and artificial intelligence. NVIDIA pioneers advancements in graphical processing units (GPUs), cloud computing, data centers, and virtual reality, with a focus on gaming, automotive, healthcare, and robotics industries. The company's innovations, such as NVIDIA Omniverse, transform traditional digital processes by enabling high-fidelity simulations and rendering tasks. Their applications span various industries, from autonomous vehicles using NVIDIA DRIVE to healthcare solutions with NVIDIA Clara, and AI-driven analytics and workflows.

📋 Description

• 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

🎯 Requirements

• Bachelor’s degree (or equivalent experience) plus 5 or more years of relevant experience in large-scale machine learning, deep learning, and general data science; or a Master’s degree or PhD plus 3 or more years of relevant experience • 3 or more years of experience with PyTorch • 2 or more years of experience training enterprise-scale machine learning models across distributed infrastructure • 2 or more years of experience designing and operating efficient training and inference workflows on GPU infrastructure, including profiling, scaling, orchestration, and resource utilization • Excellent C++ programming and software design skills • Proven experience developing, debugging, and optimizing high-performance applications, preferably with GPU acceleration using CUDA • Strong collaboration, communication, and documentation habits • Experience developing or deploying Graph Neural Network solutions using PyTorch Geometric, or a similar framework • Experience working with data warehouse and lakehouse platforms, such as Snowflake or Databricks • Experience in two or more of the following domains: finance, cybersecurity, government or national laboratories, and retail • Strong understanding of system architecture, CPU, GPU, memory, and storage systems, as well as performance optimization • Experience with customer engagement and technical support, particularly for data science workflows and with vector search and storage solutions, such as FAISS or Milvus

🏖️ Benefits

• Equity • Benefits