# Senior AI Software Engineer, Kernel Libraries

**Company**: NVIDIA
**Location**: Santa Clara
**Work arrangement**: remote
**Experience**: senior
**Job type**: full-time
**Category**: Engineering
**Industry**: Technology

**Apply**: https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-AI-Software-Engineer--Kernel-Libraries_JR2014705?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_2cab1c8b-1e8

## Description

We're looking for outstanding AI systems engineers to develop groundbreaking technologies in the inference systems software stack. As a member of the team, you'll develop libraries, code generators, and GPU kernel technologies for NVIDIA's hardware architecture.

Your key responsibilities will include:

- Innovating and developing new AI systems technologies for efficient inference

- Designing, implementing, and optimising kernels for high-impact AI workloads

- Designing and implementing extensible abstractions for LLM serving engines

- Building efficient just-in-time domain-specific compilers and runtimes

- Collaborating closely with other engineers at NVIDIA across deep learning frameworks, libraries, kernels, and GPU arch teams

- Contributing to open-source communities like FlashInfer, vLLM, and SGLang

You will also be eligible for equity and benefits.

## Skills

### Required
- PyTorch
- JAX
- TensorFlow
- ONNX
- vLLM
- SGLang
- Python
- C/C++

### Nice to have
- FlashInfer
- Flash Attention
- Apache TVM
- MLIR
- CUDA C/C++
- cuTile
- Triton

---

Source: [Apply at nvidia.wd5.myworkdayjobs.com](https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-AI-Software-Engineer--Kernel-Libraries_JR2014705?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)