ML Model Bringup Engineer for CSX AI Accelerator

Cerebras; Toronto; On-site; CAD 90,000 - 130,000; Full time

Job summary

Cerebras Systems seeks a versatile engineer for its Inference Core Model Bringup team. You will be responsible for the end-to-end implementation of open-source and proprietary ML models using Cerebras CSX systems. The role requires a strong background in Python, C/C++, debuggin across multiple levels of the AI stack, and experience with modern deep learning frameworks. Join us at the forefront of AI technology, and help us achieve unprecedented performance and scalability in AI applications.

Benefits

Competitive salary and benefits package

Opportunities for professional growth and career advancement

Dynamic and innovative work environment

Chance to work on cutting-edge technologies

Qualifications

• Strong debugging skills across performance, numerical accuracy, and runtime integration.
• Comfort navigating the full AI toolchain: Python modeling code, compiler IRs, performance profiling, etc.
• Experience with deep learning frameworks and familiarity with model internals.

Responsibilities

• Contribute to the end-to-end bring up of 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

Python

C/C++

Debugging

Deep learning frameworks (e.g., PyTorch, TensorFlow)

Compiler development (LLVM, MLIR)

Optimization techniques

Education

Bachelor’s, Master’s, or PhD in Computer Science, Engineering, or a related field

Job description

Cerebras Systems seeks a versatile engineer for its Inference Core Model Bringup team. You will be responsible for the end-to-end implementation of open-source and proprietary ML models using Cerebras CSX systems. The role requires a strong background in Python, C/C++, debuggin across multiple levels of the AI stack, and experience with modern deep learning frameworks. Join us at the forefront of AI technology, and help us achieve unprecedented performance and scalability in AI applications.