Senior Software Engineer

Job Summary

To support the development of large-scale machine learning solutions, the full-time Senior Software Engineer will create accelerated, PyTorch-based models, including Graph Neural Networks, while collaborating with cross-functional teams and providing technical leadership in a remote or onsite environment.

Key responsibilities

• Develop and optimize PyTorch-based solutions for large-scale machine learning models, focusing on GPU infrastructure
• Support CUDA-X Libraries and collaborate with teams to innovate GNN models and implementations
• Provide mentorship to engineers and identify opportunities for codebase improvements and re-architecture

Required qualifications

• Bachelor's degree (or equivalent experience) with 5+ years in large-scale machine learning or a Master's/PhD with 3+ years of relevant experience
• 3+ years of experience with PyTorch and 2+ years training enterprise-scale machine learning models on distributed infrastructure
• 2+ years designing efficient training and inference workflows on GPU infrastructure
• Excellent C++ programming and software design skills with experience in high-performance applications
• Proven ability to develop, debug, and optimize applications, preferably with GPU acceleration using CUDA