Full Stack LLM Engineer

Cerebras Systems

8 - 12 years

8 - 12 Lacs

bengaluru

Posted:5 days ago| Platform:

Apply

Skills Required

computer science performance tuning automation prototype architecture machine learning healthcare hardware open source python

Work Mode

Work from Office

Job Type

Full Time

Job Description

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 8 to 12 years experience.
Comfort navigating the full AI toolchain: Python modelling code, compiler IRs, performance profiling, etc.
Strong debugging skills across performance, numerical accuracy, and runtime integration.
Experience with deep learning frameworks (e.g., PyTorch, TensorFlow) and familiarity with model internals (e.g., attention, MoE, diffusion).
Proficiency in C/C++ programming and experience with low-level optimization.
Strong background in optimization techniques, particularly those involving NP-hard problems.

More Jobs at Cerebras Systems

Simulator Engineer

Bengaluru

3.0 - 6.0 yrs

INR 6 - 10 Lacs

QA Engineer

bengaluru

3.0 - 8.0 yrs

INR 5 - 10 Lacs

SOC Physical Design Integration Lead

bengaluru

10.0 - 15.0 yrs

INR 35 - 40 Lacs

Detection and Response Engineer

bengaluru

3.0 - 5.0 yrs

INR 5 - 7 Lacs

Senior Security Engineer - Product Security

bengaluru

5.0 - 10.0 yrs

INR 7 - 12 Lacs

Mock Interview

Practice Video Interview with JobPe AI

Start Machine Learning Interview

Start Your Job Search Today

Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.