[Skip to main content](#main-content) [ OpenNash ](https://opennash.com/) / [Jobs](https://jobs.opennash.com/) [ 01 Search jobs](https://jobs.opennash.com/search/)[ 02 Companies](https://jobs.opennash.com/explore/)[ 03 Hiring report](https://jobs.opennash.com/reports/hiring/)[ 04 AI assistants](https://jobs.opennash.com/connect/)[ 05 How it works](https://jobs.opennash.com/methodology/) Amazon · IL, Tel Aviv # Machine Learning Performance Engineer, Annapurna Labs IL, Tel Aviv Data & Analytics Listed on Amazon’s official careers site. Posted October 26, 2026. Check that it is still open and apply there. [Apply on Amazon’s careers site →](https://www.amazon.jobs/en/jobs/10589075/machine-learning-performance-engineer-annapurna-labs) [OpenNash Jobs](https://jobs.opennash.com/) / [Amazon](https://jobs.opennash.com/companies/amazon/) / Machine Learning Performance Engineer, Annapurna Labs ## Job description The Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and generative AI workloads on Amazon's custom machine learning accelerators — Inferentia and Trainium. These chips power workloads for thousands of AWS customers, from large language model training runs to real-time inference serving billions of daily predictions. We are building the first Neuron performance engineering team in Tel Aviv. As a Machine Learning Performance Engineer, you'll help shape the direction of this team from the ground up — profiling and optimizing workloads across the full ML software stack, writing high-performance kernels, and improving the Neuron SDK that external developers depend on. You'll work at the boundary between software and hardware, collaborating directly with compiler, runtime, and chip design engineers to close performance gaps customers care about. The team is new and small, which means broad scope, direct ownership, and real influence over the technical direction we take. If you enjoy digging into performance bottlenecks and turning analysis into measurable wins, this role is for you. ### Key job responsibilities Design and implement high-performance compute kernels for ML operations, leveraging the Neuron architecture and programming models. Profile ML workloads end-to-end to identify bottlenecks — memory, compute, or communication — and drive optimizations through to a measured improvement. Enhance the programming model and tooling that kernel and model developers rely on, improving usability and debugging workflows. Identify and drive optimization opportunities across the Neuron software stack (compiler, runtime, frameworks). Document software designs, operational runbooks, and performance findings so the broader team can build on your work. ### A day in the life You might start your morning reviewing profiling data from a customer's large diffusion model training job, tracing a utilization gap back to a specific kernel. After a design discussion with compiler engineers about a new operator fusion strategy, you spend the afternoon writing and benchmarking a kernel prototype. Later, you review a teammate's pull request for a runtime optimization and share your findings in a short write-up for the broader Neuron organization. Your work directly translates into faster model execution and lower cost for AWS customers running ML workloads at scale. ### About the team The Neuron Performance Engineering team in Tel Aviv is part of Annapurna Labs within AWS. Our mission is to make sure every ML workload running on Inferentia and Trainium chips reaches its full performance potential. We partner closely with compiler, framework, and hardware teams across Annapurna Labs, and we work directly with AWS customers to understand their models and unblock their adoption. We are a newly formed group which is part of the larger Neuron organization. If you want to shape a team's technical culture from its earliest days while working on problems that matter to the future of AI infrastructure, we'd love to hear from you. ## At a glance - Posted by the employer: Oct 26, 2026 - First seen by OpenNash: Nov 1, 2026 - Last checked: Nov 8, 2026 - Listed for: 13 days - Location: IL, Tel Aviv[Apply on Amazon’s careers site →](https://www.amazon.jobs/en/jobs/10589075/machine-learning-performance-engineer-annapurna-labs) ## Hiring at Amazon - Open jobs: 21,930 - New in the last 7 days: After two weekly collections - Tracked since: Jul 10, 2026 [All Amazon jobs](https://jobs.opennash.com/companies/amazon/) · [Weekly hiring report](/reports/hiring/) ## More jobs at Amazon ### [Data Center Chief Engineer, Data Center Engineering Operation](https://jobs.opennash.com/companies/amazon/jobs/8b1e42fac0746af5226a/) IL, Tel Aviv ### [Data Center Site HR Partner, AWS HRP, BeXT AIS](https://jobs.opennash.com/companies/amazon/jobs/1daf77966064b3b0e0c1/) IL, Tel Aviv ### [Data Center Technician](https://jobs.opennash.com/companies/amazon/jobs/6bbca41039b50b66b342/) IL, Tel Aviv ### [Data Scientist, Prime Video Sports](https://jobs.opennash.com/companies/amazon/jobs/c565609650dda03fbc65/) IL, Tel Aviv ### [Sr GTM SSA Analytics, EMEA GTM Data and AI Specialists Solutions Architecture](https://jobs.opennash.com/companies/amazon/jobs/70b9c63846d60eaf0966/) IL, Tel Aviv ## Similar jobs at other companies ### [Machine Learning Framework, Complier & Performance Engineer](https://jobs.opennash.com/companies/qualcomm/jobs/b1ba015fa399e1532e47/) Qualcomm · Markham, Ontario, Canada ### [Staff Machine Learning Framework, Complier & Performance Engineer](https://jobs.opennash.com/companies/qualcomm/jobs/bb9edec8f7d89d1adc3d/) Qualcomm · Canada - Remote, Canada ### [Machine Learning Engineer](https://jobs.opennash.com/companies/accenture/jobs/ef14ea91e804ba52f5de/) Accenture ### [Quantitative / Machine Learning Engineer](https://jobs.opennash.com/companies/totalenergies/jobs/259a76019d5c03765816/) TotalEnergies ### [Senior Machine Learning Engineer](https://jobs.opennash.com/companies/siemens/jobs/d151b9ceb25367914ae6/) Siemens OpenNash lists jobs from employers’ own careers sites and does not handle applications. Dates come from the employer where stated; “first seen” and “last checked” are our own observations. [ OpenNash ](https://opennash.com/) AI-native services firm that builds custom software and 24/7 AI agents. OpenNash Jobs is an OpenNash Applied AI project. [Book time with OpenNash](https://calendly.com/opennash/new-meeting?hide_gdpr_banner=1) Jobs - [Search jobs](https://jobs.opennash.com/search/) - [Companies](https://jobs.opennash.com/explore/) - [Hiring report](https://jobs.opennash.com/reports/hiring/) - [AI assistants](https://jobs.opennash.com/connect/) - [How it works](https://jobs.opennash.com/methodology/) OpenNash - [opennash.com](https://opennash.com/) - [Services](https://opennash.com/ai-agent-consulting/) - [OpenNash CX](https://opennash.com/cx/) - [AI Tools](https://opennash.com/ai-tools/) - [Blog](https://opennash.com/blog/) - [About](https://opennash.com/about/) Contact - [hello@opennash.com](mailto:hello@opennash.com) - [LinkedIn](https://www.linkedin.com/company/opennash) - [Support](https://jobs.opennash.com/support/) - [Privacy](https://jobs.opennash.com/privacy/) - [Terms](https://jobs.opennash.com/terms/) © 2026 OpenNash, Inc. Company names and public job postings belong to their respective owners. Employers control source availability and application decisions.