Описание
Job description: We are seeking an experienced AI Software Engineer to lead and contribute to our ONNX model optimization and model deployment infrastructure. You will work at the cutting edge of machine learning systems, bridging the gap between high-level deep learning frameworks and high-performance hardware execution through MLIR (Multi-Level Intermediate Representation) and ONNX (Open Neural Network Exchange) technologies.
Responsibilities
1. ) 2. Build and maintain end-to-end compilation pipelines from high-level frameworks to target hardware 3.
Implement advanced optimization passes including loop transformations, memory layout optimizations, and operator fusion 4. Collaborate with hardware teams to define and implement target-specific code generation strategies 5. Lead ONNX model import/export, conversion, and optimization initiatives 6.
Develop and maintain ONNX converters for internal frameworks and external tools 7. Implement ONNX graph optimizations, constant folding, and shape inference enhancements 8. Familiarity with deep learning model architectures (Transformers, CNNs, diffusion models) and their computational patterns 9.
Experience with hardware accelerators (TPUs, NPUs, AI engines) and their compiler stacks 10. Architect unified pipelines connecting ONNX frontend with MLIR-based compiler backends 11. Implement quantization-aware transformation
Контакты работодателя (email/phone/telegram) скрыты из публичного превью —
отправьте резюме, чтобы мы связали вас напрямую.