Voltai ยท 2 months ago
Research Engineer - CUDA Kernel Engineering
Palo Alto Office
Full-time
Onsite
Mid, Senior LevelVoltai is developing world models and embodied agents to enhance interaction with the physical world, focusing on hardware and semiconductor design. The role involves developing and optimizing CUDA kernels to support AI models for semiconductor design and verification, enabling efficient large-scale model training and inference across GPUs.
Computer Software
Responsibilities
Develop, integrate, and optimize state-of-the-art CUDA kernels to power AI models that accelerate semiconductor design and verification
Enable large-scale model training, inference, and reinforcement learning systems that reason about circuit layouts, generate and validate RTL, and optimize chip architectures
Build tools, performance benchmarks, and integration layers that push the limits of GPU utilization for compute-intensive workloads in AI-driven hardware design
Release kernels and tooling as contributions to the open-source AI and HPC ecosystems
Qualification
Required
Writing and optimizing CUDA kernels for large-scale AI workloads (attention, routing, graph-based operations, physics-inspired operators, etc.)
Profiling and optimizing GPU performance for custom compute or memory-bound workloads
Integrating custom kernels into cutting-edge training and inference frameworks (e.g., PyTorch, Megatron, vLLM, TorchTitan)
Working with the latest NVIDIA hardware and software stacks (Hopper, Blackwell, NVLink, NCCL, Triton)
Building GPU-accelerated primitives for graph reasoning, symbolic computation, or hardware simulation tasks
Collaborating with AI researchers and semiconductor experts to translate domain-specific workloads into high-performance GPU code
Company
Voltai
AI models for electronics
11-50 employees
H1B Sponsorship
Voltai has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (4)
2024 (2)
Funding
Current Stage
Early StageCompany data provided by crunchbase