Cerebras has developed a radically new chip and system to dramatically accelerate deep learning applications. Our system runs training and inference workloads orders of magnitude faster than contemporary machines, fundamentally changing the way ML researchers work and pursue AI innovation.

We are innovating at every level of the stack – from chip, to microcode, to power delivery and cooling, to new algorithms and network architectures at the cutting edge of ML research. Our fully integrated system delivers unprecedented performance because it is built from the ground up for deep learning workloads.

Responsibilities:

• Build performance models to understand the project the performance of state of the art and customer models
• Optimize our Kernel micro code and Compiler algorithms to elevate ML model utilization on the Cerebras WSE
• Debug and understand runtime performance on the system and cluster
• Design performance features for upcoming ML architectures to enable highest performance execution on both training and inference
• Develop tools and infrastructure to help visualize performance data collected from the Wafer Scale Engine and our compute cluster

Requirements:

• Masters in Electrical Engineering or Computer Science
• Strong background in computer architecture
• Strong analytical and problem solving mindset
• 3+ years of experience in a relevant domain (Computer Architecture, Network Performance, CPU/GPU Performance, Kernel Optimization, HPC)
• Experience working on CPU/GPU simulators
• Exposure to performance profiling and debug on any system pipeline
• Comfort with C++ and Python
• Exposure to and basic understanding of machine learning is desired
• Any pre-silicon performance validation exposure is a plus but not necessary

Why Join Cerebras

People who are serious about software make their own hardware. At Cerebras we have built a breakthrough architecture that is unlocking new opportunities for the AI industry. With dozens of model releases and rapid growth, we’ve reached an inflection point in our business. Members of our team tell us there are five main reasons they joined Cerebras:

1. Build a breakthrough AI platform beyond the constraints of the GPU
2. Publish and open source their cutting-edge AI research
3. Work on one of the fastest AI supercomputers in the world
4. Enjoy job stability with startup vitality
5. Our simple, non-corporate work culture that respects individual beliefs

Read our blog: Five Reasons to Join Cerebras in 2024.

Apply today and become part of the forefront of groundbreaking advancements in AI.