Want to build the future compute platform for AGI?t
Want to build hardware and software to train and run the largest ML workloads for AGI?
We are currently looking for Kernel Engineers to join a company focusing on becoming THE compute platform for AGI.
The company has over $80 million in Series A funding and are enhancing AI by building powerful hardware, specifically tailored for Large Language Models (LLMs).
They have a top-tier founding team spinning out of Google’s PaLM and with their hardware, individual researchers can train 7B-class models from scratch every day, and 70B-class models multiple times per month.
As a Kernel Engineer, you will design and optimize high-performance kernels that directly interface with proprietary hardware, shaping the compute platform for AGI. Collaborate with ML Research and Hardware teams to integrate cutting-edge software and hardware solutions for the largest ML workloads.
Requirements
- Bachelor of Computer Science or equivalent degree
- Experience optimizing software for specialized hardware, employing techniques such as parallelism, SIMD programming, C, assembly-level optimization, or GPU/CUDA programming
- Experience in Assembly, C++, C, Zig, or Rust
This role is hybrid in the Bay Area