# ML Systems Performance Engineer

**Company:** [Cerebras Systems](https://hotfix.jobs/companies/cerebras-systems)
**Location:** Bengaluru, India
**Role:** ML Engineering
**Experience:** 3+ years
**Skills:** C++, Python, Computer Architecture, Deep Learning, LLMs, Kernel Optimization, Compiler Algorithms, Cpu/Gpu Performance, High-Performance Computing, Cpu/Gpu Simulators, Performance Profiling, Performance Modeling, Wafer Scale Engine
**Posted:** 2026-06-16

> Optimizes end-to-end machine learning inference performance across kernels, compilers, systems, and clusters. The role requires strong computer architecture knowledge, experience with performance modeling and profiling, and proficiency in C++ and Python.

## Job Description

## Responsibilities
- Build performance models at the kernel and end-to-end levels to estimate performance for state-of-the-art and customer machine learning models.
- Optimize and debug kernel microcode and compiler algorithms to improve machine learning model inference speed, throughput, and compute utilization on the Cerebras Wafer Scale Engine.
- Debug and understand runtime performance on the system and cluster.
- Develop tools and infrastructure to visualize performance data collected from the Wafer Scale Engine and compute cluster.

## Requirements
- Bachelor's, master's, or PhD in Electrical Engineering or Computer Science.
- Strong background in computer architecture.
- Understanding of low-level deep learning and large language model mathematics.
- Strong analytical and problem-solving skills.
- 3+ years of experience in a relevant domain, such as computer architecture, CPU/GPU performance, kernel optimization, or high-performance computing.
- Experience working with CPU/GPU simulators.
- Experience with performance profiling and debugging on a system pipeline.
- Proficiency with C++ and Python.

## Benefits
- Work on a breakthrough AI platform beyond the constraints of GPUs.
- Publish and open-source cutting-edge AI research.
- Work on a high-performance AI supercomputer.
- Startup vitality with job stability.
- A non-corporate work culture that respects individual beliefs.

## Similar jobs

- [Research Software Engineer, Post Training](https://hotfix.jobs/jobs/168e3c8f-8577-4482-bf94-91b3d11744ba) - Thinking Machines Lab - San Francisco, CA - $350k – $475k/yr
- [Software Engineer, AI for Chip Design](https://hotfix.jobs/jobs/abfb017d-1ce2-4d24-901d-16084eb7b3bc) - OpenAI - San Francisco, CA - $266k – $468k/yr
- [AI Software Engineer](https://hotfix.jobs/jobs/c9a0e889-a36e-488f-87d0-e9c27ab63bf0) - Rollstack - Remote
- [Machine Learning Engineer, Ranking & Retrieval](https://hotfix.jobs/jobs/adb5411d-8bfc-48b8-a8e7-15cbf5ea21b2) - ClickUp - Remote - $200k – $250k/yr
- [Machine Learning Engineer III](https://hotfix.jobs/jobs/2359be26-5005-4fa7-94c9-8a86066a6bb5) - PathAI - Boston, MA - $131k – $200k/yr

**Apply:** https://hotfix.jobs/jobs/0987eef1-33c2-4eb9-884b-75ed628a2858
**Canonical:** https://hotfix.jobs/jobs/0987eef1-33c2-4eb9-884b-75ed628a2858