Kernel Engineer(m/w/x)

Beschreibung

You will drive the company's performance by designing robust GPU kernels that solve complex memory challenges, ensuring seamless high-speed operations across training and inference systems.

Anforderungen

• •Deep expertise in computer architecture
• •Low-level machine optimizations expertise
• •Code generation expertise
• •Breadth across machine learning
• •Low-level programming for AI accelerators
• •GPU kernel development and optimization
• •Experience in kernel authoring frameworks
• •Agility, ownership mindset, and grit
• •Quick learning and grit

Aufgaben

• •Design and implement high-performance GPU kernels
• •Optimize kernels for training and inference throughput
• •Manage kernel design, deployment, and production reliability
• •Ensure robustness through extensive testing and functional correctness
• •Minimize latency and memory utilization for long-context windows
• •Evaluate porting compute kernels to alternative hardware
• •Collaborate with training, inference, and RL teams on co-design

Tools & Technologien

Benefits