Full Stack Machine Learning Engineer
Cerebras SystemsOuter Ring Road, BangalorePosted 23 days agoA free account shows the apply link for every role and matches you to the ones that fit your resume.
About the role
Cerebras builds AI accelerator hardware and systems for model training and inference. This engineer will bring up machine learning models and optimize their compilation and execution on accelerator hardware.
What you'll do
- Bring up models and verify them against reference implementations
- Develop MLIR dialects, transformations and hardware-specific lowering paths
- Optimize operators, inference execution and memory efficiency
- Profile workloads and diagnose performance bottlenecks
- Validate numerical correctness and assess model quality impacts
Skills
Must have
- C++
- Python
- PyTorch
- Mlir
- Compiler engineering
- Transformer architectures
- Attention mechanisms
- Performance optimization
- Numerical precision
- GPU profiling
Nice to have
- LLM inference
- Quantization
- Accelerator kernel development
- Distributed execution
- Model parallelism
Tools
- Llvm
Who it's for
- Domain
- ai hardware
About Cerebras Systems
Cerebras Systems has 6 open roles in Bangalore, at Cerebras Systems Bengaluru. See every Cerebras Systems role.
More at Cerebras Systems
Senior · 6 to 8 years
2 months ago
Manager · 7+ years
3 months ago
Lead or staff · 10+ years
3 months ago
Senior · 5+ years
4 months ago
Mid level · 3+ years
4 months ago
Similar roles
All ml and ai engineering jobsEl
Eli Lilly and Company · Manager · 10+ years
10 days ago
