Demo

CUDA Kernels Engineer

GenBio AI
Stanford, CA Full Time
POSTED ON 2/22/2025
AVAILABLE BEFORE 5/18/2025

Headquartered in Silicon Valley, we are a newly established start-up, where a collective of visionary scientists, engineers, and entrepreneurs are dedicated to transforming the landscape of biology and medicine through the power of Generative AI. Our team comprises leading minds and innovators in AI and Biological Science, pushing the boundaries of what is possible. We are dreamers who reimagine a new paradigm for biology and medicine.

We are committed to decoding biology holistically and enabling the next generation of life-transforming solutions. As the first mover in pan-modal Large Biological Models (LBM), we are pioneering a new era of biomedicine, with our LBM training leading to ground-breaking advancements and a transformative approach to healthcare. Our exceptionally strong R&D team and leadership in LLM and generative AI position us at the forefront of this revolutionary field. With headquarters in Silicon Valley, California, and a branch office in Paris, we are poised to make a global impact. Join us as we embark on this journey to redefine the future of biology and medicine through the transformative power of Generative AI.

Key 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 accuracyWork 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

Job Requirement

  • 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 formatsHave 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 1-3 years working experience or Engineer with M.S. with 3-5 year working experience
  • Preferred Qualifications

  • 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
  • Join us as we embark on this journey to redefine the future of biology and medicine.

    We are an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees.

    If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
    Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

    What is the career path for a CUDA Kernels Engineer?

    Sign up to receive alerts about other jobs on the CUDA Kernels Engineer career path by checking the boxes next to the positions that interest you.
    Income Estimation: 
    $85,996 - $102,718
    Income Estimation: 
    $111,859 - $131,446
    Income Estimation: 
    $110,457 - $133,106
    Income Estimation: 
    $105,809 - $128,724
    Income Estimation: 
    $122,763 - $145,698
    Income Estimation: 
    $151,448 - $188,145
    Income Estimation: 
    $203,425 - $249,816
    Income Estimation: 
    $213,375 - $267,876
    Income Estimation: 
    $190,687 - $235,769
    Income Estimation: 
    $85,996 - $102,718
    Income Estimation: 
    $111,859 - $131,446
    Income Estimation: 
    $110,457 - $133,106
    Income Estimation: 
    $105,809 - $128,724
    Income Estimation: 
    $122,763 - $145,698
    Income Estimation: 
    $105,809 - $128,724
    Income Estimation: 
    $136,611 - $163,397
    Income Estimation: 
    $135,163 - $163,519
    Income Estimation: 
    $131,953 - $159,624
    Income Estimation: 
    $150,859 - $181,127
    Income Estimation: 
    $73,784 - $86,677
    Income Estimation: 
    $90,372 - $103,622
    Income Estimation: 
    $61,825 - $80,560
    Income Estimation: 
    $90,032 - $105,965
    Income Estimation: 
    $85,996 - $102,718
    View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

    Job openings at GenBio AI

    GenBio AI
    Hired Organization Address CA Full Time
    Headquartered in Silicon Valley, we are a newly established start-up, where a collective of visionary scientists, engine...
    GenBio AI
    Hired Organization Address Stanford, CA Full Time
    Headquartered in Silicon Valley, we are a newly established start-up, where a collective of visionary scientists, engine...

    Not the job you're looking for? Here are some other CUDA Kernels Engineer jobs in the Stanford, CA area that may be a better fit.

    Software Engineer - Kernels

    MatX, Mountain View, CA

    Senior Software Engineer - CUDA

    Snarkify, Stanford, CA

    AI Assistant is available now!

    Feel free to start your new journey!