Skip to content
VoltaiVoltaiPalo Alto, CA

Research Engineer - CUDA Kernel Engineering

Develops and optimizes CUDA kernels for AI models accelerating semiconductor design, verification, and chip optimization across large-scale GPU clusters. Integrates with frameworks like PyTorch and latest NVIDIA hardware for training, inference, and RL workloads.

Salary not listed
On-siteBackend Engineering

About the role

Responsibilities

  • Develop, integrate, and optimize state-of-the-art CUDA kernels to power AI models accelerating semiconductor design and verification.
  • Enable large-scale model training, inference, and reinforcement learning systems for circuit layouts, RTL generation/validation, and chip architecture optimization across thousands of GPUs.
  • Build tools, performance benchmarks, and integration layers to push GPU utilization limits for compute-intensive workloads in AI-driven hardware design.
  • Collaborate with researchers and engineers; contribute kernels and tooling to open-source AI and HPC ecosystems.

Requirements

  • Writing and optimizing CUDA kernels for large-scale AI workloads (attention, routing, graph-based operations, physics-inspired operators).
  • Profiling and optimizing GPU performance for custom compute or memory-bound workloads.
  • Integrating custom kernels into training and inference frameworks (PyTorch, Megatron, vLLM, TorchTitan).
  • Working with latest NVIDIA hardware/software (Hopper, Blackwell, NVLink, NCCL, Triton).
  • Building GPU-accelerated primitives for graph reasoning, symbolic computation, or hardware simulation.
  • Collaborating with AI researchers and semiconductor experts to translate domain-specific workloads into high-performance GPU code.

Skills

CUDAPyTorchNvidia HopperNvidia BlackwellNvlinkNcclTriton Inference ServerMegatronvLLMTorchtitan

Similar roles

Backend Engineering jobs
Front

Software Engineer (Backend), Deliverability

FrontSan Francisco, CA

Technical lead on the Deliverability team owning high-volume email infrastructure, ingestion, routing, and outbound delivery systems that process millions of events. Requires 5+ years building scalable distributed backend systems with strong ownership of architecture, reliability, and on-call duties.

178k – 204k/yr
Hybrid5+ YOEBackend Engineering
Scribe

Backend Engineer

ScribeSan Francisco, CA

Backend Engineer building and optimizing core document workflows (viewing, editing, embeds, exports) and greenfield projects in a Django monolith. Requires 3+ years backend experience, strong Python skills, performance optimization, and automated testing.

140k – 185k/yr
Hybrid3+ YOEBackend Engineering
OpenAI

Software Engineer, Codex -Enterprise Controls

OpenAISan Francisco, CA

Build and operate the enterprise control plane for OpenAI's Codex AI software engineer, focusing on security, governance, identity, access controls, encryption, and policy enforcement systems for large organizations.

230k – 385k/yr
Hybrid5+ YOEBackend Engineering
ZoomInfo

Software Engineer III

ZoomInfoBethesda, MD +1

Software Engineer III designing, building, and maintaining scalable backend systems with Java, Node.js/NestJS, Kafka, and cloud technologies (AWS/GCP). Requires Master's in CS + 2 years (or Bachelor's + 4 years) backend experience, microservices, databases, and REST APIs.

112k – 176k/yr
Hybrid4+ YOEBackend Engineering
Upstart

Software Engineer, Upstart Bank

UpstartUnited States

Software Engineer building scalable backend systems, services, and APIs for Upstart Bank, a modern AI-enabled national bank. Translate regulatory and business requirements into reliable software in a fintech environment; 4+ years backend/full-stack experience required.

14k – 187k/yr
Remote4+ YOEBackend Engineering