Full Stack LLM Engineer

About the role

We are looking for a versatile engineer to join the Inference Core Model Bringup team at Cerebras. The role focuses on rapidly bringing state-of-the-art open-source and proprietary LLMs onto Cerebras CSX systems, driving unprecedented performance and efficiency.

Key responsibilities

• Contribute to end-to-end model bring-up on Cerebras CSX hardware.
• 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 tooling, API, or automation improvements to accelerate future bring-ups.

Required profile

• Bachelor’s, Master’s or PhD in Computer Science, Engineering or a related field.
• Comfort navigating the full AI toolchain, including Python modeling code, compiler IRs and performance profiling.
• Strong debugging skills for performance, numerical accuracy and runtime integration.
• Experience with deep-learning frameworks such as PyTorch or TensorFlow and familiarity with model internals (attention, MoE, diffusion).
• Proficiency in C/C++ programming and low-level optimization.
• Proven experience in compiler development, especially with LLVM or MLIR.
• Background in optimization techniques, including NP-hard problem solving.

Required skills

• Python
• C / C++
• PyTorch
• TensorFlow
• LLVM
• MLIR
• Performance profiling
• Compiler optimization

What we offer

• Competitive salary and benefits package.
• Opportunities for professional growth and career advancement.
• A dynamic, innovative work environment.
• Chance to work on cutting-edge AI technologies that shape the future of the industry.