Senior AI Kernel & Inference Engineer

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

Job summary

A leading technology company is seeking a Senior AI Software Engineer to join their team in Santa Clara, California. In this role, you will innovate and develop groundbreaking AI systems software for inference applications including deep learning framework optimizations and GPU kernel technologies. You will closely collaborate with other engineers to enhance NVIDIA’s leading technology and ensure the efficiency of AI workloads. Ideal candidates will possess a Master's degree in Computer Science, strong programming skills in C/C++, and significant experience with ML frameworks.

Qualifications

• 6+ years of experience in ML/DL systems development preferred.
• Strong experience with GPU kernel development.
• PhD preferred for advanced roles.

Responsibilities

• Innovate and develop new AI systems technologies.
• Design and optimize kernels for AI workloads.
• Collaborate with other engineers on deep learning frameworks.

Skills

Deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX)

C/C++ programming

Machine Learning/Deep Learning systems development

Education

Master's degree in Computer Science or Electrical Engineering

Tools

CUDA C/C++

Apache TVM

MLIR

Job description

A leading technology company is seeking a Senior AI Software Engineer to join their team in Santa Clara, California. In this role, you will innovate and develop groundbreaking AI systems software for inference applications including deep learning framework optimizations and GPU kernel technologies. You will closely collaborate with other engineers to enhance NVIDIA’s leading technology and ensure the efficiency of AI workloads. Ideal candidates will possess a Master's degree in Computer Science, strong programming skills in C/C++, and significant experience with ML frameworks.