At a Glance
- Tasks: Optimise AI model performance using advanced compiler technologies and profiling techniques.
- Company: Join a leading tech firm in Edinburgh focused on AI innovation.
- Benefits: Competitive salary, flexible working hours, and opportunities for professional growth.
- Other info: Dynamic team environment with a focus on research and development.
- Why this job: Make a real impact in the AI field while working with cutting-edge technology.
- Qualifications: Experience with LLVM/MLIR and AI frameworks like TensorFlow or PyTorch.
The predicted salary is between 49500 - 60500 £ per year.
We are seeking a skilled AI Compiler Optimization Engineer to optimize AI model inference performance through advanced compiler technologies. You will focus on performance tuning for CPU or hybrid CPU/XPU heterogeneous architectures, profiling AI frameworks to discover new optimization opportunities, and delivering cutting-edge insights from industry research.
Key Responsibilities:
- Compiler-Based Performance Optimization: Implement compiler techniques (e.g., MLIR level optimizations, LLVM backend optimizations) to enhance inference performance on CPU and CPU/XPU hybrid systems. Optimize JIT level compute graphs with operator fusion, memory allocation and etc. for latency/throughput improvements. Preferred: Experience with LLVM/MLIR development.
- AI Model Profiling & Framework Optimization: Profile end-to-end inference workflows on frameworks like TensorFlow, PyTorch, ONNX, and llama.cpp to identify hotspots and bottlenecks. Propose and implement optimization strategies (e.g., kernel tuning, graph-level optimizations). Preferred: Experience optimizing models on multiple AI frameworks.
- Research & Insight Development: Track and analyze the latest advancements in AI & compiler research (academic papers, open-source projects). Produce actionable insight reports summarizing trends, benchmarks, and potential optimizations. Preferred: Strong technical writing skills with prior publications or reports.
AI Compiler Optimization Engineer - Edinburgh in London employer: microTECH Global LTD
Join a forward-thinking company in Edinburgh that values innovation and collaboration, making it an excellent employer for an AI Compiler Optimization Engineer. With a strong focus on employee growth, you will have access to cutting-edge projects and the opportunity to work alongside industry experts, fostering a culture of continuous learning and development. Enjoy a supportive work environment that encourages creativity and offers competitive benefits, ensuring your contributions are recognised and rewarded.
