Kernel Engineer

Cerebras Systems builds the world's largest AI chip, 56 times larger than GPUs, delivering industry-leading training and inference speeds for machine learning applications.
Embedded
Mid-Level Software Engineer
In-Person
3+ years of experience
AI · Hardware

Description For Kernel Engineer

Cerebras Systems, a pioneering company in AI hardware, is seeking a Kernel Engineer to join their innovative team in Bengaluru. The company builds the world's largest AI chip, which is 56 times larger than traditional GPUs, revolutionizing AI compute capabilities.

As a Kernel Engineer, you'll work at the fascinating intersection of hardware and software, developing state-of-the-art solutions for emerging problems in AI and HPC. You'll be part of a team responsible for designing, implementing, validating, and performance-tuning deep learning operations on highly parallel custom processors.

The role involves developing sophisticated kernel libraries using low-level assembly instructions and domain-specific languages, optimizing algorithms for the Cerebras hardware system, and working closely with chip architects to enhance next-generation systems. You'll also contribute to the evolution of kernel library architecture to address computational challenges in cutting-edge Neural Networks.

Cerebras offers a unique opportunity to work with industry leaders, publish research, and contribute to groundbreaking advancements in AI technology. The company maintains a simple, non-corporate culture that respects individual beliefs while providing the stability of an established organization with the energy of a startup.

The ideal candidate should have a strong background in computer science or related fields, expertise in C++ and Python, and a deep understanding of hardware architecture. Experience with kernel development, parallel algorithms, and machine learning frameworks would be highly valuable. This is an excellent opportunity for someone passionate about pushing the boundaries of AI computing and working with cutting-edge technology.

Last updated 14 days ago

Responsibilities For Kernel Engineer

  • Develop design specifications for new machine learning and linear algebra kernels
  • Develop and debug kernel library of highly optimized low level assembly instruction and C-like domain specific language routines
  • Use mathematical models and analysis to measure software performance
  • Develop and integrate unit and system testing methodologies
  • Study emerging trends in Machine Learning applications
  • Interact with chip and system architects to optimize instruction sets

Requirements For Kernel Engineer

Python
Linux
  • Bachelor's, Master's, PhD or foreign equivalents in Computer Science, Computer Engineering, Mathematics, or related fields
  • Understanding of hardware architecture concepts
  • Skilled in C++ and Python programming languages
  • Good knowledge of library and/or API development best practices
  • Strong debugging skills and knowledge of debugging complex software stack
  • Experience in kernel development and/or testing (preferred)
  • Familiarity with parallel algorithms and distributed memory systems (preferred)
  • Experience in programming accelerators such as GPUs and FPGAs (preferred)
  • Familiarity with Machine Learning neural networks and frameworks (preferred)
  • Familiarity with HPC kernels and their optimization (preferred)

Benefits For Kernel Engineer

  • Work on one of the fastest AI supercomputers in the world
  • Opportunity to publish and open source cutting-edge AI research
  • Job stability with startup vitality
  • Simple, non-corporate work culture

Interested in this job?

Jobs Related To Cerebras Systems Kernel Engineer

Software Development Engineer II, Device Software

Software Development Engineer II position at Amazon working on device software for consumer electronics like Kindle, Fire TV, and Echo devices.

Embedded Engineer, Graviton Team

Embedded Engineer position at Annapurna Labs (AWS) working on the Graviton team to develop next-generation cloud computing platforms and silicon devices.

Software Development Engineer

Amazon Lab126 is seeking an Embedded Software Development Engineer to design and develop next-generation devices, working with kernel drivers and BSP integration.

Electrical Test and Reliability Engineer (Starlink)

Electrical Test and Reliability Engineer position at SpaceX working on Starlink satellite hardware testing and reliability engineering in Redmond, WA.

Electrical Test and Reliability Engineer (Starlink)

Electrical Test and Reliability Engineer position at SpaceX's Starlink division, focusing on satellite hardware reliability and automated testing systems.