Kernel Engineer - High-Performance ML/HPC on Custom AI Chip

Cerebras Systems; United States; On-site; USD 110,000 - 140,000; Full time

Benefits offered by this job

Non-corporate work culture

Equal opportunity employer

Continuous learning and growth opportunities

Job summary

Cerebras Systems is seeking a Kernel Engineer to develop high-performance software solutions at the intersection of hardware and software. You'll implement and optimize deep learning operations to leverage a custom, massively parallel processor architecture. Join a world-class team to enhance AI innovations.

This role involves designing kernel libraries, collaborating on architectural optimizations, and staying updated with ML trends. Ideal candidates are skilled in C++ and Python, with a strong educational background in relevant fields.

Cerebras promotes diversity and offers a unique work environment that empowers individuals to excel.

Qualifications

• Understanding of hardware architecture concepts.
• Comfortable learning new hardware architecture details.
• Good knowledge of library and/or API development.

Responsibilities

• Develop design specifications for new ML and linear algebra kernels.
• Develop and debug kernel library of optimized low-level assembly instructions.
• Using mathematical models to measure software performance.

Skills

C++

Python

Debugging skills

Education

Bachelor’s, Master’s, PhD in Computer Science, Computer Engineering, or Mathematics

Tools

TensorFlow

PyTorch

Job description

Cerebras Systems is seeking a Kernel Engineer to develop high-performance software solutions at the intersection of hardware and software. You'll implement and optimize deep learning operations to leverage a custom, massively parallel processor architecture. Join a world-class team to enhance AI innovations.

This role involves designing kernel libraries, collaborating on architectural optimizations, and staying updated with ML trends. Ideal candidates are skilled in C++ and Python, with a strong educational background in relevant fields.

Cerebras promotes diversity and offers a unique work environment that empowers individuals to excel.