LLM Inference Performance & Evals Engineer
Toronto, Ontario, Canada
⚠️ We'll shut down after Aug 1st - try foo🦍 for all jobs in tech ⚠️
Cerebras Systems
Cerebras is the go-to platform for fast and effortless AI training. Learn more at cerebras.ai.Cerebras Systems builds the world's largest AI chip, 56 times larger than GPUs. Our novel wafer-scale architecture provides the AI compute power of dozens of GPUs on a single chip, with the programming simplicity of a single device. This approach allows Cerebras to deliver industry-leading training and inference speeds and empowers machine learning users to effortlessly run large-scale ML applications, without the hassle of managing hundreds of GPUs or TPUs.
Cerebras' current customers include global corporations across multiple industries, national labs, and top-tier healthcare systems. In January, we announced a multi-year, multi-million-dollar partnership with Mayo Clinic, underscoring our commitment to transforming AI applications across various fields. In August, we launched Cerebras Inference, the fastest Generative AI inference solution in the world, over 10 times faster than GPU-based hyperscale cloud inference services.
About The Role
Join the inference model team dedicated to bring up the state-of-the-art models, numerically validating and accelerating new model ideas on wafer-scale hardware. You will prototype architectural tweaks, build performance-eval pipelines, and turn hard numbers into changes that land in production.
Key Responsibilities
- Prototype and benchmark cutting-edge ideas: new attentions, MoE, speculative decoding, and many more innovations as they emerge.
- Develop agent-driven automation that designs experiments, schedules runs, triages regressions, and drafts pull-requests.
- Work closely with compiler, runtime, and silicon teams: unique opportunity to experience the full stack of software/hardware innovation.
- Keep pace with the latest open- and closed-source models; run them first on wafer scale to expose new optimization opportunities.
Skills And Qualifications
- 3 + years building high-performance ML or systems software.
- Solid grounding in Transformer math—attention scaling, KV-cache, quantisation—or clear evidence you learn this material rapidly.
- Comfort navigating the full AI toolchain: Python modeling code, compiler IRs, performance profiling, etc.
- Strong debugging skills across performance, numerical accuracy, and runtime integration.
- Prior experience in modeling, compilers or crafting benchmarks or performance studies; not just black-box QA tests.
- Strong passion to leverage AI agents or workflow orchestration tools to boost personal productivity.
Assets
- Hands-on with flash-attention, Triton kernels, linear-attention, or sparsity research.
- Performance-tuning experience on custom silicon, GPUs, or FPGAs.
- Proficiency in C/C++ programming and experience with low-level optimization.
- Proven experience in compiler development, particularly with LLVM and/or MLIR.
- Publications, repos, or blog posts dissecting model speed-ups.
- Contributions to open-source agent frameworks.
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:
- Build a breakthrough AI platform beyond the constraints of the GPU.
- Publish and open source their cutting-edge AI research.
- Work on one of the fastest AI supercomputers in the world.
- Enjoy job stability with startup vitality.
- Our simple, non-corporate work culture that respects individual beliefs.
Read our blog: Five Reasons to Join Cerebras in 2025.
Apply today and become part of the forefront of groundbreaking advancements in AI!
Cerebras Systems is committed to creating an equal and diverse environment and is proud to be an equal opportunity employer. We celebrate different backgrounds, perspectives, and skills. We believe inclusive teams build better products and companies. We try every day to build a work environment that empowers people to do their best work through continuous learning, growth and support of those around them.
This website or its third-party tools process personal data. For more details, click here to review our CCPA disclosure notice.
* Salary range is an estimate based on our AI, ML, Data Science Salary Index 💰
Tags: Architecture Generative AI GPU LLMs Machine Learning Mathematics Open Source Pipelines Python Research
Perks/benefits: Career development Startup environment
More jobs like this
Explore more career opportunities
Find even more open roles below ordered by popularity of job title or skills/products/technologies used.