Research Engineer - CUDA Kernel Engineering

The role

You will develop, integrate, and optimize CUDA kernels to power AI models for semiconductor design and verification. You will build tools, performance benchmarks, and integration layers to push GPU utilization limits.

What you will do

• Develop, integrate, and optimize state-of-the-art CUDA kernels
• Enable large-scale model training, inference, and reinforcement learning systems
• Build tools, performance benchmarks, and integration layers
• Release kernels and tooling as contributions to open-source AI and HPC ecosystems
• Collaborate with researchers and engineers

Nice to have

• Experience writing and optimizing CUDA kernels for large-scale AI workloads
• Experience profiling and optimizing GPU performance for custom compute or memory-bound workloads
• Experience integrating custom kernels into cutting-edge training and inference frameworks
• Experience working with the latest NVIDIA hardware and software stacks
• Experience building GPU-accelerated primitives for graph reasoning, symbolic computation, or hardware simulation tasks
• Experience collaborating with AI researchers and semiconductor experts

About the company

Voltai is developing world models and agents to learn, evaluate, plan, experiment, and interact with the physical world. The company is starting with understanding and building hardware, electronics systems and semiconductors.