Kernel Engineer - New Grad
Core
Develop high-performance software at the intersection of hardware and software for cutting-edge artificial intelligence and high-performance computing workloads.
Role type
New Grad Kernel Engineer (AI/HPC)
Builds
Machine learning and linear algebra operations for the Cerebras Wafer-Scale Engine
Domain
Artificial Intelligence / High-Performance Computing / Custom Hardware
Deliverable
production ML models
Required skills
C++, Python, computer architecture, data structures, algorithms, debugging, parallel programming, performance optimization, low-level software, hardware/software co-design
Preferred skills
kernel development, compilers, HPC, systems programming, parallel algorithms, multithreaded programming, distributed memory systems, GPU/FPGA programming, CUDA, OpenCL, domain-specific languages, machine learning concepts, neural networks, PyTorch, TensorFlow, numerical computing, profiling tools, library development
Technologies
Cerebras Software Language, Cerebras Wafer-Scale Engine, PyTorch, TensorFlow
Responsibilities
Design and implement machine learning and linear algebra kernels; Develop and debug high-performance kernel routines using low-level programming techniques; Apply parallel programming algorithms to map computational workloads; Evaluate kernel behavior using mathematical analysis and profiling tools; Investigate correctness, performance, and hardware utilization issues; Develop unit tests and system-level validation methodologies; Collaborate with kernel, compiler, performance, and hardware engineers; Study emerging machine learning workloads; Participate in code reviews and technical discussions; Build an understanding of the Cerebras architecture, instruction set, memory system, and communication model.
Seniority
New Grad, hands-on IC
