CUDA Kernels Engineer

Company: Genbio
Location: Palo Alto
Posted on: April 12, 2025

Job Description:

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 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.Job Requirements
• 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.
• 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 years 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.
  #J-18808-Ljbffr

Keywords: Genbio, Hayward , CUDA Kernels Engineer, Engineering , Palo Alto, California