Senior Software Engineer, AI and DL Kernel Libraries

Full-stack EngineerSoftware EngineerFull TimeRemoteTeam 10,001+Since 1993H1B SponsorCompany SiteLinkedIn

Location

California + 3 moreAll locations: California, Colorado, Texas, Washington

Posted

2 days ago

Salary

$184K - $287.5K / year

Postgraduate Degree6 yrs expEnglishOpen SourcePythonPy TorchTensorflow

Job Description

• Innovating and developing new AI systems technologies for efficient inference • Designing, implementing, and optimizing 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

Job Requirements

  • Masters degree in Computer Science, Electrical Engineering, or related field (or equivalent experience); PhD are preferred
  • 6+ years (academic/ industry) experience with ML/DL systems development preferable
  • Strong experience in developing or using deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX, etc) and ideally inference engines and runtimes such as vLLM, SGLang, and MLC.
  • Strong Python and C/C++ programming skills
  • Strong experience in GPU kernel development and performance optimizations (especially using CUDA C/C++, cuTile, Triton, or similar)

Benefits

  • equity
  • benefits

Related Job Pages

More Full-stack Engineer Jobs

Senior Staff Software Engineer

DDN

World’s leading Data Intelligence Platform supercharging over 500,000 GPUs across all data workloads

Full-stack Engineer2 days ago
Full TimeRemoteTeam 1,001-5,000Since 1998H1B Sponsor

Senior Software Engineer leading scalable storage solutions for AI and data management

Distributed SystemsJavaKubernetesOpenStackRustSwiftGo
United States
Full-stack Engineer2 days ago
Full TimeRemoteTeam 5,001-10,000

The Senior Fullstack Engineer will drive the design, architecture, and implementation of high-performance features across the entire stack, from Python microservices to React/TypeScript frontends. This role involves leading frontend development, building robust backend APIs, managing cloud infrastructure, and providing technical mentorship to junior engineers.

United States

Senior Software Development Engineer

CVS Health

Bringing our heart to every moment of your health.

Full-stack Engineer2 days ago
Full TimeRemoteTeam 10,001+Since 1963H1B No Sponsor

This role involves designing, developing, and implementing digital solutions by applying advanced technical expertise to architect and code software applications, conduct testing, and contribute to the overall technical direction of engineering projects. Key duties include defining technical architecture, writing complex code, collaborating on high-level architectural decisions, and ensuring the scalability and performance of digital solutions.

United States + 2 moreAll locations: United States, Bulgaria, United Kingdom
$83.4K - $222K / year

Senior Full Stack Engineer

LeadSimple

The #1 CRM, workflow automation, & shared inbox software for property managers. Manage more doors with less stress.

Full-stack Engineer2 days ago
Full TimeRemoteTeam 51-200Since 2013H1B No Sponsor

At LeadSimple, we build software that helps property managers and real estate professionals streamline operations and build strong customer relationships. We’re looking for a Senior Full-Stack Engineer to build and maintain our web applications using React on the frontend and R...

United States