> Markdown version of [/jobs/ext/2965559-sr-ml-kernel-performance-engineer-aws-neuron-annapurna-labs-c-150-700-c-251-700-a-year](https://www.wearedevelopers.com/jobs/ext/2965559-sr-ml-kernel-performance-engineer-aws-neuron-annapurna-labs-c-150-700-c-251-700-a-year). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# Sr. Ml Kernel Performance Engineer, Aws Neuron, Annapurna Labs - C$150,700 - C$251,700 A Year

- **Company:** Amazon.com, Inc.
- **Location:** Madrid, Spain
- **Contract:** Internship / Graduate position
- **Skills:** Computer-Aided Design, Adobe InDesign, Amazon Web Services, Code Review, Nvidia CUDA, Computer Programming, Shard (Database Architecture), Microprocessors, Field-Programmable Gate Array (FPGA), General-Purpose Computing on Graphics Processing Units, Integrated Development Environments, Machine Learning, OpenCL, Performance Tuning, Tensorflow, Software Engineering, Graphics Processing Unit (GPU), Pytorch, Deep Learning, Parallel Computation, Backend, Information Technology, Build Process, Machine Learning Operations, Software Coding, Software Version Control
- **Published:** September 17, 2026
- **Apply:** https://www.buscojobs.com.es/sr-ml-kernel-performance-engineer-aws-neuron-annapurna-labs-c-150-700-c-251-700-a-year-en-madrid-ID-371696532

## About the Role

decisions with technical input.Operate in a startup?like development environment, solving the most important problems.Qualifications5+ years of professional software development experience (non?internship).5+ years of programming in at least one software programming language.5+ years of leading design or architecture of scalable, reliable systems.Experience as a mentor, technical lead, or engineering team leader.Full software development life cycle experience, including coding standards, code reviews, source control, build processes, testing, and operations.Bachelor's degree in computer science or equivalent (preferred).
Expertise in accelerator architectures for ML or HPC such as GPUs, CPUs, FPGAs, or custom architectures.Experience with GPU kernel optimization and GPGPU computing (CUDA, NKI, Triton, OpenCL, SYCL, ROCm).
Proficiency in low?level performance optimization and memory hierarchy understanding.Experience developing high?performance libraries for HPC applications.Knowledge of ML frameworks (PyTorch, TensorFlow) and their GPU backends.Experience with LLVM/MLIR backend development for GPUs.Experience with parallel programming and optimization techniques.Understanding of GPU memory hierarchies and optimization strategies.Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.Base salary range: CAN, ON, Toronto - 150,************,****** CAD annually.
Compensation may include sign?on payments, restricted stock units (RSUs), and other elements.

## Description

Sr.ML Kernel Performance Engineer, AWS Neuron, Annapurna LabsThe Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and GenAI workloads on Amazon's custom machine learning accelerators, Inferentia and Trainium.The Acceleration Kernel Library team focuses on maximizing performance for AWS's custom ML accelerators, crafting high-performance kernels for ML functions at the hardware?software boundary.Key Job ResponsibilitiesDesign and implement high-performance compute kernels for ML operations leveraging the Neuron architecture and programming models.Analyze and optimize kernel-level performance across multiple generations of Neuron hardware.Conduct detailed performance analysis using profiling tools to identify and resolve bottlenecks.Implement compiler optimizations such as fusion, sharding, tiling, and scheduling.Work directly with customers to enable and optimize their ML models on AWS accelerators.Collaborate across teams to develop innovative kernel optimization techniques.A Day in the LifeBuild high-impact solutions for a global customer base.Participate in design discussions, code reviews, and communicate with internal and external stakeholders.Work cross?functionally to help drive business decisions with technical input.Operate in a startup?like development environment, solving the most important problems.Qualifications5+ years of professional software development experience (non?internship).5+ years of programming in at least one software programming language.5+ years of leading design or architecture of scalable, reliable systems.Experience as a mentor, technical lead, or engineering team leader.Full software development life cycle experience, including coding standards, code reviews, source control, build processes, testing, and operations.Bachelor's degree in computer science or equivalent (preferred).Expertise in accelerator architectures for ML or HPC such as GPUs, CPUs, FPGAs, or custom architectures.Experience with GPU kernel optimization and GPGPU computing (CUDA, NKI, Triton, OpenCL, SYCL, ROCm).Proficiency in low?level performance optimization and memory hierarchy understanding.Experience developing high?performance libraries for HPC applications.Knowledge of ML frameworks (PyTorch, TensorFlow) and their GPU backends.Experience with LLVM/MLIR backend development for GPUs.Experience with parallel programming and optimization techniques.Understanding of GPU memory hierarchies and optimization strategies.Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.Base salary range: CAN, ON, Toronto - 150,************,****** CAD annually.Compensation may include sign?on payments, restricted stock units (RSUs), and other elements.Amazon offers comprehensive benefits including health insurance, a Registered Retirement Savings Plan (RRSP), Deferred Profit Sharing Plan (DPSP), paid time off, and resources to improve health and well?being.Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.#J-*****-Ljbffr


## Related Videos

- [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding)
- [Accelerating Python on GPUs](https://www.wearedevelopers.com/videos/859-accelerating-python-on-gpus)
- [Developing the Backend with Stefan Lingler, CTO at Shpock](https://www.wearedevelopers.com/videos/100360-developing-the-backend-with-stefan-lingler-cto-at-shpock)
- [Nest.js - TypeScript in the backend can also be clean](https://www.wearedevelopers.com/videos/1033-nest-js-typescript-in-the-backend-can-also-be-clean)
- [Serverless deployment of (large) NLP models ](https://www.wearedevelopers.com/videos/158-serverless-deployment-of-large-nlp-models)
- [Coffee with Developers - Stephen Jones - NVIDIA](https://www.wearedevelopers.com/videos/1303-coffee-with-developers-stephen-jones-nvidia)

## Related Articles

- [Highest Paying Tech Companies for Developers](https://www.wearedevelopers.com/magazine/220-highest-paying-tech-companies-for-developers)
- [Highest Paying Tech Companies in Europe](https://www.wearedevelopers.com/magazine/162-highest-paying-tech-companies-in-europe)
- [Software Developer Salary in Germany [2023]](https://www.wearedevelopers.com/magazine/194-software-developer-salary-in-germany-2023)
- [Backend Developer Salary in Germany [2023]](https://www.wearedevelopers.com/magazine/196-backend-developer-salary-in-germany-2023)
- [Fullstack developer salary in Germany [2023]](https://www.wearedevelopers.com/magazine/197-fullstack-developer-salary-in-germany-2023)
- [Software Engineer Salary London](https://www.wearedevelopers.com/magazine/252-software-engineer-salary-london)