CUDA Kernels Engineer

Posted 8 Days Ago
Be an Early Applicant
Palo Alto, CA
Mid level
Artificial Intelligence • Software
The Role
Develop GPU/CPU kernels for high performance, optimize AI architecture training, collaborate on model design, and work with hardware vendors.
Summary Generated by Built In

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.

Top Skills

Ai Architectures
C/C++
Cuda
Low-Precision Formats
Python
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Palo Alto, CA
29 Employees
On-site Workplace
Year Founded: 2024

What We Do

GenBio.AI, Inc. (GenBio AI) is an innovative global startup dedicated to developing the world's first AI-driven Digital Organism, an integrated system of multiscale foundation models for predicting, simulating, and programming biology at all levels.

Our goal is to achieve comprehensive, actionable empirical understandings of the mechanisms underlying all organismal physiologies and diseases. This will pave the way for a new paradigm in drug design, bio-engineering, personalized medicine, and fundamental biomedical research, all powered by Generative Biology.

Our founding team consists of world-renowned scientists and researchers in AI and Biology from prestigious institutions such as CMU, MBZUAI, WIS, alongside prominent financial investors.

GenBio AI, a true global effort from day one, is establishing offices in Palo Alto, Paris, and Abu Dhabi.

Similar Jobs

Capital One Logo Capital One

Manager, Software Engineering, Full Stack (People Leader)

Fintech • Machine Learning • Payments • Software • Financial Services
Hybrid
2 Locations
55000 Employees
211K-241K Annually

Crusoe Energy Systems Logo Crusoe Energy Systems

Staff Network Engineer, Deployment

Cloud • Greentech • Other • Energy
Hybrid
San Francisco, CA, USA
667 Employees
195K-230K Annually

Crusoe Energy Systems Logo Crusoe Energy Systems

Senior+ Network Engineer, Deployment

Cloud • Greentech • Other • Energy
Hybrid
San Francisco, CA, USA
667 Employees
175K-205K Annually

Atlassian Logo Atlassian

Principle Machine Learning Engineer -- Teamwork Graph

Cloud • Information Technology • Productivity • Security • Software • App development • Automation
Remote
Mountain View, CA, USA
11000 Employees
190K-306K Annually

Similar Companies Hiring

True Anomaly Thumbnail
Software • Machine Learning • Hardware • Defense • Artificial Intelligence • Aerospace
Colorado Springs, CO
131 Employees
Caliola Engineering Thumbnail
Software • Machine Learning • Hardware • Defense • Data Privacy • App development • Aerospace
Colorado Springs, CO
53 Employees
Red 6 Thumbnail
Virtual Reality • Software • Hardware • Defense • Aerospace
Orlando, Florida
113 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account