GenBio AI ยท 1 week ago
CUDA Kernels Engineer
Palo Alto, CA
Full-time
Onsite
Mid, Senior Level
$170K/yr - $260K/yr
1+ years expGenBio AI is a newly established start-up headquartered in Silicon Valley, focused on transforming biology and medicine through Generative AI. The CUDA Kernels Engineer will develop high-performance GPU/CPU kernels, optimize algorithms, and collaborate with platform teams to enhance training and inference capabilities.
Artificial Intelligence (AI)BiotechnologyMedical
Hiring Manager
Bilige S.
Responsibilities
Develop high-performance GPU/CPU kernels and make trade-offs that maximize end-to-end hardware utilization
Utilize knowledge of hardware features and performance characteristics to make aggressive optimizations
Work with our other platform teams to deploy your kernels, manage our training uptime, and find opportunities for optimization
Develop low-precision algorithms that deliver high performance with little loss of ML accuracy
Work with ML engineers to develop model architectures that are amenable to efficient training and inference
Work with hardware vendors and advise on HW/SW co-design
Qualification
Required
A strong coder with excellent skills in C/C++ and Python
Have a deep understanding of GPU, CPU, or other AI accelerator architectures
Have experience writing and optimizing compute kernels with CUDA or similar languages
Are familiar with LLM (and/or other foundation model) architectures and training infrastructure
Have experience driving ML accuracy with low-precision formats
Have 1+ years of relevant industry experience
Get a great deal of satisfaction with every percentage point in performance improvement
Senior Engineer for Ph.D. with 2-3 years working experience or Engineer with M.S. with 3-5 year working experience
Preferred
PhD in Computer Science and Engineering with a specialization in Computer Architecture, Parallel Computing, Compilers, or other Systems
Experience building compilers
Experience working with hardware developers
Company
GenBio AI
GenBio AI creates AI-driven models to simulate and predict biological systems at multiple scales.
11-50 employees
H1B Sponsorship
GenBio AI 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 (3)
2024 (1)
Funding
Current Stage
Early StageRecent News
2025-11-14
Company data provided by crunchbase