{"schemaVersion":"jobsearcher.job.v1","id":"61e4f435dd672fc362f27baf","url":"https://jobsearcher.com/jobs/61e4f435dd672fc362f27baf","canonicalUrl":"https://jobsearcher.com/jobs/61e4f435dd672fc362f27baf","title":"Machine Learning Performance Engineer, Annapurna Labs","description":"Overview\nIn this role you will help build and optimize ML performance across the AWS Neuron stack for Inferentia and Trainium. You will work with cross-functional teams in Tel Aviv to push end-to-end system performance, kernel optimization, and SDK enhancements. You’ll tackle challenging problems across the stack and shape the direction of a new core team. This is a hands-on role at the intersection of ML and systems with meaningful impact on large-scale AI infrastructure.\n\nCompensation / Benefitsflexible work hoursmentorship and career growthinclusive team culturework-life balancediverse experiences encouragedcareer advancement resources\nResponsibilitiesSolve cross-stack performance challenges and design innovative software to improve service performance, durability, cost, and securityResearch and implement solutions to deliver optimal experiences for customersDesign, implement, test, deploy, and maintain high-performance software across the ML stackDevelop high-impact solutions for a large customer base and contribute to end-to-end performance gainsParticipate in design discussions and code reviews; communicate with internal and external stakeholdersWork cross-functionally to inform technical decisions and drive business outcomesOperate in a startup-like environment focusing on the most important work\nKey requirementsB.S. in Computer Science (or related field)Proficiency in Python or C++ (Python preferred)Experience with TensorFlow, PyTorch, and/or JAX3+ years of non-internship software development experience3+ years of experience in performance optimizations for deep-learning models (LLMs, vision, etc.)cross-functional collaborationstrong problem-solving mindseteffective communication with stakeholdersPythonC++TensorFlow","company":"Amazon","rawCompany":"amazon","city":"Chicago","state":"IL","isRemote":false,"isActive":false,"createdAt":"2026-09-15T03:23:16.540Z","occupations":[{"code":"15-1252.00","title":"Software Developers","slug":"software-developers"},{"code":"15-1299.08","title":"Computer Systems Engineers/Architects","slug":"computer-systems-engineers-architects"},{"code":"15-2051.00","title":"Data Scientists","slug":"data-scientists"}],"industries":[{"code":"541511","title":"Custom Computer Programming Services","slug":"custom-computer-programming-services"},{"code":"541512","title":"Computer Systems Design Services","slug":"computer-systems-design-services"},{"code":"513210","title":"Software Publishers","slug":"software-publishers"}],"jobPosting":{"@context":"https://schema.org","@type":"JobPosting","title":"Machine Learning Performance Engineer, Annapurna Labs","description":"Overview\nIn this role you will help build and optimize ML performance across the AWS Neuron stack for Inferentia and Trainium. You will work with cross-functional teams in Tel Aviv to push end-to-end system performance, kernel optimization, and SDK enhancements. You’ll tackle challenging problems across the stack and shape the direction of a new core team. This is a hands-on role at the intersection of ML and systems with meaningful impact on large-scale AI infrastructure.\n\nCompensation / Benefitsflexible work hoursmentorship and career growthinclusive team culturework-life balancediverse experiences encouragedcareer advancement resources\nResponsibilitiesSolve cross-stack performance challenges and design innovative software to improve service performance, durability, cost, and securityResearch and implement solutions to deliver optimal experiences for customersDesign, implement, test, deploy, and maintain high-performance software across the ML stackDevelop high-impact solutions for a large customer base and contribute to end-to-end performance gainsParticipate in design discussions and code reviews; communicate with internal and external stakeholdersWork cross-functionally to inform technical decisions and drive business outcomesOperate in a startup-like environment focusing on the most important work\nKey requirementsB.S. in Computer Science (or related field)Proficiency in Python or C++ (Python preferred)Experience with TensorFlow, PyTorch, and/or JAX3+ years of non-internship software development experience3+ years of experience in performance optimizations for deep-learning models (LLMs, vision, etc.)cross-functional collaborationstrong problem-solving mindseteffective communication with stakeholdersPythonC++TensorFlow","datePosted":"2026-09-15T03:23:16.540Z","dateModified":"2026-09-15T03:23:16.540Z","hiringOrganization":{"@type":"Organization","name":"Amazon","sameAs":"https://jobsearcher.com"},"jobLocation":{"@type":"Place","address":{"@type":"PostalAddress","addressLocality":"Chicago","addressRegion":"IL","addressCountry":"US"}},"identifier":{"@type":"PropertyValue","name":"JobSearcher","value":"61e4f435dd672fc362f27baf"},"url":"https://jobsearcher.com/jobs/61e4f435dd672fc362f27baf"}}