CUDA Kernel Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Periodic Labs · 2 months ago

CUDA Kernel Engineer

Periodic Labs is an AI + physical sciences lab focused on making novel scientific discoveries. The role involves developing, integrating, and optimizing CUDA kernels to enhance AI scientific research and support large-scale experiments.

Artificial Intelligence (AI)Foundational AIGenerative AIMachine Learning
check
H1B Sponsor Likelynote

Responsibilities

Develop, integrate and optimize state-of-the-art CUDA kernels to power AI scientific research
Integrate CUDA kernels into training, inference and reinforcement learning systems running on thousands of GPUs
Build tools and directly support frontier-scale experiments to make Periodic Labs the world’s best AI + science lab
Release kernels as contributions to the open-source AI stack

Qualification

CUDA kernelsNvidia hardwareInference frameworksTraining frameworksReinforcement learning

Required

Writing and optimizing CUDA kernels: attention, mixture-of-experts, dispatch-and-combine, and others
Working with the latest generation of Nvidia hardware
Integrating kernels into state-of-the-art inference (vLLM, SGLang) and training frameworks (Megatron, TorchTitan)

Company

Periodic Labs

twittertwittertwitter
company-logo
Periodic Labs develops artificial intelligence systems that simulate and predict the properties of materials using machine learning.

H1B Sponsorship

Periodic Labs 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 (2)

Funding

Current Stage
Early Stage
Total Funding
$300M
2025-09-30Seed· $300M
Company data provided by crunchbase