Lead Full Stack Machine Learning Engineer
Core
Rapidly bring up state-of-the-art open-source models, frameworks, and data engineering for Cerebras CSX systems to achieve unprecedented performance and scalability for AI applications.
Role type
Lead Full Stack Machine Learning Engineer (System-minded generalist)
Builds
End-to-end bring up of frameworks for RL, inference serving, and ML models on Cerebras CSX systems
Domain
AI hardware acceleration / Deep learning frameworks / Compiler optimization
Deliverable
production ML models
Required skills
Python modelling, compiler IRs, performance profiling, deep learning frameworks (PyTorch, TensorFlow), C/C++ programming, low-level optimization, NP-hard problem optimization
Preferred skills
Model internals knowledge (attention, MoE, diffusion), system-minded generalist approach
Technologies
Cerebras CSX, PyTorch, TensorFlow, C/C++, Python
Responsibilities
Contribute to end-to-end bring up of frameworks for RL, inference serving, and ML models; Work across the stack including 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
Seniority
Senior, hands-on IC
