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.
Backend
Mid-Level Software Engineer
In-Person
501 - 1,000 Employees
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 team in Bengaluru, India. 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 parallel and distributed algorithms to maximize hardware utilization and accelerate neural network training to unprecedented speeds. You'll work with low-level assembly instructions and C-like domain-specific language routines, while also collaborating with chip and system architects to optimize next-generation systems.

The ideal candidate should have a strong academic background in Computer Science, Computer Engineering, or Mathematics, with expertise in C++ and Python programming. Knowledge of hardware architecture, debugging complex software stacks, and API development best practices is essential. Experience with kernel development, parallel algorithms, GPU programming, and machine learning frameworks would be highly advantageous.

Cerebras offers a unique opportunity to work on groundbreaking AI technology, with the chance to publish research and contribute to one of the world's fastest AI supercomputers. The company combines the stability of an established organization with the dynamic environment of a startup, fostering a non-corporate culture that respects individual beliefs and promotes continuous learning and growth.

Last updated 15 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
  • Using 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
  • 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 QA Engineer, Vega OS

Quality Assurance Engineer role at Amazon Lab126, focusing on test automation and quality leadership for consumer electronics devices and OS development.

Software Development Engineer II, MENA Tech

Amazon MENA is hiring a Software Development Engineer II to design and build innovative software solutions in Amman, Jordan.

Support Engineer II, Payments Acceptance and Experience Tech

Support Engineer II role at Amazon's Payments team, combining technical support with development work in Python and Java, focusing on payment systems and customer experience.

Support Engineer II, Payments Acceptance and Experience Tech

Support Engineer role at Amazon focusing on Payments Acceptance and Experience Tech, requiring 2+ years experience in software development or technical support.

Software Dev Engineer II, ESTech

Software Development Engineer II position at Amazon's Amazonian Experience and Technology team, focusing on building human capital management technology and employee experience solutions.