Senior AI Inference Kernel Engineer

NVIDIA Gruppe; Santa Clara (CA); On-site; USD 184,000 - 287,500; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA Gruppe in Santa Clara is seeking an AI Systems Engineer to innovate and develop cutting-edge technologies in the AI inference software stack. Candidates should hold a Master's degree and possess over 6 years of experience in ML/DL systems development.

The role involves building efficient kernels and compilers for AI workloads while actively contributing to open-source projects. The offering includes a competitive base salary ranging from $184,000 to $287,500, with eligibility for equity and benefits.

Qualifications

• Masters degree in Computer Science, Electrical Engineering, or related field; PhD preferred.
• 6+ years of experience with ML/DL systems development.
• Strong experience in developing or using deep learning frameworks.

Responsibilities

• Innovating and developing new AI systems technologies for efficient inference.
• Designing, implementing, and optimizing kernels for high impact AI workloads.
• Collaborating closely with other engineers at NVIDIA.

Skills

Deep learning frameworks experience

Python programming

C/C++ programming

GPU kernel development

Education

Masters degree in Computer Science or related field

Tools

PyTorch

JAX

TensorFlow

Job description

NVIDIA Gruppe in Santa Clara is seeking an AI Systems Engineer to innovate and develop cutting-edge technologies in the AI inference software stack. Candidates should hold a Master's degree and possess over 6 years of experience in ML/DL systems development.

The role involves building efficient kernels and compilers for AI workloads while actively contributing to open-source projects. The offering includes a competitive base salary ranging from $184,000 to $287,500, with eligibility for equity and benefits.