Senior Software Engineer, GNN

What you'll do

• Develop accelerated PyTorch-based solutions for large-scale machine learning models like GNNs and Tabular Foundation Models.
• Optimize training and inference workflows for high-performance execution on NVIDIA GPU infrastructure.
• Support CUDA-X libraries and integrations within large-scale machine learning workflows.
• Gather technical requirements from customers and Solutions Architects to guide product and engineering priorities.
• Provide technical leadership and mentorship to engineers across the software development team.
• Identify opportunities to improve the codebase and reduce maintenance overhead through system re-architecture.
• Apply agentic coding tools to identify bugs, implement new features, and refactor existing code.
• Solve complex technical issues and coordinate across multiple teams to achieve shared project objectives.

What we're looking for

• Bachelor’s degree plus experience in large-scale machine learning, deep learning, and data science, or a Master’s/PhD plus .
• 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.
• Excellent C++ programming and software design skills.
• Proven experience developing, debugging, and optimizing high-performance applications, preferably with GPU acceleration using CUDA (preferred).
• Experience developing or deploying Graph Neural Network solutions using PyTorch Geometric or similar frameworks (preferred).
• Experience with data warehouse and lakehouse platforms such as Snowflake or Databricks (preferred).