ML Compiler Engineer

About the Role

Cohere is hiring an ML Compiler Engineer to optimize model inference through compiler techniques. You will develop custom optimizations for transformer models, implement efficient kernel generation, and work on cross-platform model compilation.

Requirements

• 5+ years of compiler or systems engineering experience
• Strong C++ and Python skills
• Experience with LLVM, MLIR, or similar frameworks
• Understanding of GPU architectures and parallel computing
• Experience with model compilation or optimization

Nice to Have

• Experience with Triton or custom CUDA kernels
• Background in auto-tuning or performance optimization
• Familiarity with quantization-aware compilation
• Experience with multiple hardware backends (NVIDIA, AMD, custom)

Benefits

Skills

Related Jobs