Senior Software Engineer, AI and DL Kernel Libraries

Job not on LinkedIn

🕒 March 12

🏄 California, Colorado, +2 more states – Remote

infoinfo

💵 $184k - $287.5k / year

⏰ Full Time

🟠 Senior

🤖 AI Engineer

🦅 H1B Visa Sponsor

infoinfo

👻 Ghost score 36%

infoinfo
Apply Now
Find Similar Remote Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Logo of NVIDIA

NVIDIA

10,000+ employees

Founded 1993

🏥 Healthcare

🏭 Manufacturing

🤖 Artificial Intelligence

Healthcare • Manufacturing • Artificial Intelligence

NVIDIA is a leading technology company specializing in accelerated computing and artificial intelligence. NVIDIA pioneers advancements in graphical processing units (GPUs), cloud computing, data centers, and virtual reality, with a focus on gaming, automotive, healthcare, and robotics industries. The company's innovations, such as NVIDIA Omniverse, transform traditional digital processes by enabling high-fidelity simulations and rendering tasks. Their applications span various industries, from autonomous vehicles using NVIDIA DRIVE to healthcare solutions with NVIDIA Clara, and AI-driven analytics and workflows.

📋 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

🎯 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

Apply Now

Similar Jobs

🕒 March 9

Alteryx

1001 - 5000

💼 Consulting

🏥 Healthcare

🏭 Manufacturing

Senior GTM Strategy Lead at Alteryx focusing on AI and cloud data strategies. Driving cross-functional initiatives to accelerate revenue through strategic partnerships and technical growth.

🕒 February 27

Andromeda

11 - 50

🏥 Healthcare

💼 Consulting

🏨 Hospitality

Infrastructure Product Engineer at Andromeda, building scalable AI infrastructure and collaborating with cross-functional teams. Play a key role in transforming complex infrastructure challenges into robust product capabilities.

🇺🇸 United States – Remote

🔥 Funding within the last year

💰 $15.1M Series A - Andromeda Robotics on 2025-09

⏰ Full Time

🟡 Mid-level

🟠 Senior

🤖 AI Engineer

🦅 H1B Visa Sponsor

infoinfo

🕒 February 27

Zillion Technologies, Inc.

501 - 1000

💼 Consulting

🏥 Healthcare

📦 Logistics

AI Architect leading design and deployment of an AI-powered bot for streamlining billing workflows. Collaborating with teams to enhance citizen-facing services and ensure compliance.

🕒 February 26

Epistemix

11 - 50

💼 Consulting

🏥 Healthcare

📣 Marketing

AI Engineer designing, developing, and deploying AI-driven applications at Epistemix. Contributing to the product roadmap and company growth with innovative solutions.

🇺🇸 United States – Remote

💰 $7M Series A - Epistemix on 2024-06

⏰ Full Time

🟡 Mid-level

🟠 Senior

🤖 AI Engineer

🕒 February 26

Tribe AI

51 - 200

💼 Consulting

🏥 Healthcare

🤖 Artificial Intelligence

Forward Deployed AI Architect designing complex AI systems in partnership with enterprises. Leading technical discovery and client engagement to achieve significant AI outcomes.