Senior AI Compiler Engineer
Develop and optimize AI graph compiler software for NPU workloads, including large language models, in collaboration with compiler and hardware teams.
Обязанности
- Design and own key components of an AI graph compiler.
- Develop and optimize neural network execution flows, memory management, and data movement.
- Enable complex AI workloads, including large language models, and implement new NPU features.
- Analyze performance bottlenecks and drive system-level optimizations.
- Collaborate with compiler and hardware teams on hardware-software solutions.
Требования
- At least 5 years of software development experience using C or C++.
- BSc or MSc in Computer Science, Electrical Engineering, or an equivalent field.
- Experience designing and developing complex software systems.
- Strong understanding of memory management and performance optimization.
Будет плюсом
- Experience with AI accelerators, NPUs, GPUs, or DSPs.
- Experience with AI compilers, graph optimization, or neural network execution.
- Experience with large language models or other large neural network workloads.
- Familiarity with PyTorch, TensorFlow, ONNX, or Python.