Senior CUDA Deep Learning Engineer

Job Summary

Senior CUDA Deep Learning Engineer, a full-time position focused on integrating advanced CUDA features into AI frameworks and improving performance for large-scale deep learning applications.

Key Responsibilities

• Integrate new CUDA features and runtime abstractions into AI frameworks, from proof of concept to production
• Analyze AI workloads to identify opportunities for innovation and collaborate with teams on AI models
• Drive improvements in the AI Compiler-Runtime interface for multi-GPU solutions and design fault-tolerant systems for dynamic workloads

Required Qualifications

• BS, MS, or PhD in Computer Science, Computer Engineering, Electrical Engineering, or related field (or equivalent experience)
• 8+ years of relevant industry or academic experience post-degree
• Development experience with deep learning frameworks such as PyTorch and JAX
• Proficiency in Python, C++, CUDA, or related DSLs for rapid prototyping
• Understanding of AI models, parallelism, and performance benchmarking on AI clusters