Location

India Office
Employment Type

Full time
Department

Software

Cerebras Systems builds the world's largest AI chip, 56 times larger than GPUs. This architecture allows Cerebras to deliver industry-leading training and inference speeds; over 10 times faster than GPU-based hyperscale cloud inference services.

This order of magnitude increase in speed is transforming the user experience of AI applications, unlocking real-time iteration and increasing intelligence via additional agentic computation.

Cerebras works with the leading model labs, global enterprises, and cutting-edge AI-native startups. OpenAI recently announced a multi-year partnership with Cerebras, to deploy 750 megawatts of scale, transforming key workloads with ultra high-speed inference.

About The Role

This teams' principal responsibility is to rapidly bring up state-of-the-art open-source models, frameworks and data engineering. Success in this role requires a system-minded generalist who thrives in fast-paced bringup environments and is comfortable working across the entire software stack.



Your work will play a critical role in achieving unprecedented levels of performance, efficiency, and scalability for AI applications.

Responsibilities
- Contribute to the end-to-end bring up of frameworks for RL, inference serving, ML models on Cerebras CSX systems.
- Work across the stack: model architecture translation, graph lowering, compiler optimizations, runtime integration, and performance tuning.
- Debug performance and correctness issues spanning model code, compiler IRs, runtime behavior, and hardware utilization.
- Propose and prototype improvements across tools, APIs, or automation flows to accelerate future bring ups.

Skills & Qualifications
- Bachelor’s, Master’s, or PhD in Computer Science, Engineering, or a related field with 10+ years’ experience.
- Comfort navigating the full AI toolchain: Python modelling code, compiler IRs, performance profiling, etc.
- Solid debugging skills a

📌 Lead Full Stack Machine Learning Engineer (India)
🏢 Cerebras Systems
📍 India