Senior Software Engineer, Matrix Multiplication

Job Description

The role involves innovating and developing new AI systems technologies for efficient inference and designing, implementing, and optimizing kernels for high-impact AI workloads. Collaboration with other engineers at NVIDIA across various teams is also a key aspect of the job.

Requirements: Candidates should have a Master's degree in Computer Science, Electrical Engineering, or a related field, with a preference for a PhD. A minimum of 6 years of experience in ML/DL systems development and strong programming skills in Python and C/C++ are required.

Key Skills: Machine Learning, Deep Learning, Python, C/C++, GPU Kernel Development, Performance Optimization, Matrix Multiplication, AI Systems, Inference Engines, Domain Specific Compilers, Open Source Contributions, LLM Serving Engines, CUDA, Triton, FlashInfer, SGLang

Benefits: Equity, Health Insurance