ML Performance Benchmarking Engineer

Cerebras Systems

$90K — $130K *
Enterprise Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • Bachelor's or Master's degree in Computer Engineering, Systems Engineering, or a related field
  • Proficiency in Python and/or C++ programming
  • Proven experience in building and scaling automated infrastructure
  • Strong background in throughput and performance optimization techniques
  • Excellent problem-solving skills and a strong analytical mindset
  • Demonstrated ability to dive deep into new domains
  • Ability to work in a fast-paced, ambiguous, and collaborative environment

Responsibilities

  • Design and implement end-to-end telemetry systems for inference performance
  • Architect, build, and scale automation for performance data analysis
  • Dive deep into system behavior to identify performance bottlenecks
  • Deliver actionable insights for feature prioritization and evolution
  • Collaborate with Core Platform teams to validate inference features
  • Drive performance improvements from prototyping to production deployment

Benefits

  • Opportunity to work at the forefront of AI technology
  • Collaboration with a passionate and innovative engineering team
  • Engagement in challenging and impactful projects
  • Flexible working arrangements (on-site or hybrid)
Full Job Description
About The Role

The Inference Core Platform group is at the heart of Cerebras' mission to deliver the world's fastest AI inference. Our team builds the foundational software and hardware infrastructure that powers low-latency, high-speed, high-throughput deployment on the Cerebras Wafer-Scale Engine (WSE). We are responsible for the full stack-from model compilation and scheduling down to custom hardware kernels and driver development.

The ML Performance Benchmarking team plays a pivotal role in shaping the performance and scalability of AI inference on one of the most advanced computing systems ever built. We drive the bring-up of core inference capabilities and deliver performance improvements at every stage of development - from early prototyping to production deployment.

We're looking for passionate engineers to join us in redefining the limits of AI inference. If you thrive on building systems that measure, analyze, and optimize performance at scale, this is your opportunity to make a transformative impact on the future of AI.

Scope of the team includes:
  • Core Inference Observability - Design and implement end-to-end telemetry systems across the software stack, providing deep visibility into inference performance and enabling rapid iteration before and after deployment.
  • Benchmarking Infrastructure - Architect, build, and scale the automation that generates, analyzes, and visualizes performance data used to inform business decisions across engineering and leadership.
  • Performance Analysis - Dive deep into system behavior, dissect performance bottlenecks, and deliver actionable insights that directly influence which features ship and how they evolve.
  • Feature Integration - Partner closely with Core Platform teams to define rigorous testing methodologies that validate inference features for peak performance.

Skills & Qualifications
  • Bachelor's or Master's degree in Computer Engineering, Systems Engineering, or a related field.
  • Proficiency in Python and/or C++ programming.
  • Proven experience in building and scaling automated infrastructure.
  • Strong background in throughput and performance optimization techniques, especially in complex, large-scale systems.
  • Excellent problem-solving skills and a strong analytical mindset.
  • Demonstrated ability to dive deep into new domains.
  • Ability to work in a fast-paced, ambiguous, and collaborative environment.

Preferred Skills & Qualifications
  • Familiarity with problem-solving at the intersection of hardware and software.
  • Hands-on experience with AI workloads and architectures is a plus.

Location
  • On-site or hybrid at our Toronto office


#LI-WA1

Similar Jobs

More Jobs at Cerebras Systems

More Enterprise Technology Jobs

Find similar ML Performance Benchmarking Engineer jobs: