Kernel Engineer

Cerebras Systems

3 days ago

Sunnyvale, CA, USA or Toronto, Canada

Mid Level / Senior

H1B Sponsor

Responsibilities

Develop design specifications for new machine learning and linear algebra kernels.
Implement and debug kernel library routines using low-level assembly and a custom C-like language.
Measure software performance using mathematical models to inform design decisions.
Develop and integrate testing methodologies to verify kernel library functionality.
Study emerging trends in Machine Learning to evolve kernel library architecture.
Collaborate with chip and system architects to optimize next-generation systems.

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 stacks.

Opportunity to build a breakthrough AI platform beyond GPU constraints.
Ability to publish and open source cutting-edge AI research.
Work on one of the fastest AI supercomputers in the world.
Enjoy job stability with startup vitality.
Experience a simple, non-corporate work culture that respects individual beliefs.

C++PythonPyTorchTensorFlow