Machine Learning Performance Engineer, Annapurna LabsAI

Amazon

View & apply on the source site ↗

Company: Amazon
Location: Tel Aviv
Office: HQ · 121 Menachem Begin Rd, Tel Aviv-Yafo, Israel
Field: Data Science, ML & Algorithms
Role: ML Engineer
Type: Full Time
Posted: 8 Dec 2025

The posting date comes from source metadata.

Location

About Amazon

View company page →

Skills & technologies

AWS
GenAI
LLM
Machine Learning
PyTorch
Python
TensorFlow

Description

Our team is responsible for the AWS Neuron software stack, which powers Generative AI and other advanced ML workloads on AWS's custom-built ML accelerators — Inferentia and Trainium. These...

Requirements

B.S. or M.S. in computer science or related field - Proficiency with 1 or more of the following programming languages: Python(preferred), C++ - Experience working with TensorFlow, PyTorch, and/or JAX - 3+ years of non-internship professional software development experience - 3+ years of experience in performance optimizations in LLM, Vision or other deep-learning models - M.S. in computer science or related field - Experience with developing algorithms for simulation tools - Experience developing compiler optimization, kernel writing or hardware-software co-design Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

Listing aggregated by Goozali Jobs. Apply on the employer's site.