C

Cerebras

AI accelerator hardware replacing GPUs

Full Stack LLM Engineer

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeToronto, ON, CanadaNo salary listedHybrid

About the Job

Requirements

  • A bachelor's, master's, or PhD in Computer Science, Engineering, or a related field is required.
  • Comfort navigating the full AI toolchain, including Python modeling code, compiler intermediate representations, and performance profiling.
  • Strong debugging skills across performance, numerical accuracy, and runtime integration.
  • Experience with deep learning frameworks such as PyTorch or TensorFlow and familiarity with model internals such as attention, mixture of experts, and diffusion.
  • Proficiency in C/C++ programming and experience with low-level optimization.
  • Proven experience in compiler development, particularly with LLVM and/or MLIR.
  • A strong background in optimization techniques, particularly those involving NP-hard problems.

Responsibilities

  • Contribute to end-to-end bringup of machine learning models on Cerebras CSX systems.
  • Work across the stack, including model architecture translation, graph lowering, compiler optimizations, runtime integration, and performance tuning.
  • Debug performance and correctness issues spanning model code, compiler intermediate representations, runtime behavior, and hardware utilization.
  • Propose and prototype improvements across tools, APIs, or automation flows to accelerate future bringups.

About the Company

Cerebras

AI accelerator hardware replacing GPUs

Cerebras Systems creates AI acceleration hardware and software. Its CS-2 system is designed to replace traditional GPU clusters for AI workloads, speeding up training and inference while simplifying the setup by eliminating the need for parallel programming, distributed training, and cluster management. The product works as a single, large processor-based accelerator with accompanying software and cloud services to run AI models efficiently, reducing latency and time to results. Compared with competitors, Cerebras differentiates itself with the largest processor in the industry and an integrated hardware-software stack that aims to streamline AI workflows rather than relying on multi-GPU clusters. The company's goal is to help research labs, healthcare, finance, and other industries achieve faster, more cost-effective AI development and deployment by offering a turnkey high-performance AI compute solution.

Company size
501-1,000
Company stage
IPO
Headquarters
Sunnyvale, CA
Founded
2016

Recently Posted Jobs

Explore more opportunities at Cerebras

See more jobs