About this opportunity
NVIDIA AI lists this Senior Software Engineer, GNN opportunity in austin, California. Review the employer’s description below for duties, qualifications and application requirements.
Job description
Develop and optimize accelerated PyTorch-based solutions for large-scale machine learning models, including GNNs and ensemble models on GPU infrastructure. Collaborate with cross-functional teams to support CUDA-X libraries and guide product priorities based on customer technical requirements.
Requirements:
Requires a Bachelor's degree with 5+ years of experience or a Master's/PhD with 3+ years in machine learning and data science. Candidates must possess strong C++ skills, extensive experience with PyTorch, and a proven track record in GPU-accelerated application development.
Key Skills:
PyTorch, C++, CUDA, Graph Neural Networks, Machine Learning, Deep Learning, GPU Acceleration, Distributed Infrastructure, Data Science, Performance Optimization, Software Design, System Architecture, PyTorch Geometric, Profiling, Orchestration
Benefits:
Equity, Benefits
#J-18808-Ljbffr
Who can apply
Review the original listing for work authorization, qualifications and employer requirements.