{"schemaVersion":"jobsearcher.job.v1","id":"42c70cd957e93a42447210a1","url":"https://jobsearcher.com/jobs/42c70cd957e93a42447210a1","canonicalUrl":"https://jobsearcher.com/jobs/42c70cd957e93a42447210a1","title":"ML Distributed Training Performance Engineer - Trainium","description":"Amazon Web Services (AWS) Annapurna Labs is seeking a software engineer for the Distributed Training team in AWS Neuron to optimize large‑scale ML model training across Trainium. You will work on throughput, memory and compiler/runtime performance across the Neuron stack.\nYou will collaborate with PyTorch and JAX teams and push improvements through the Neuron compiler and runtime, enabling efficient training on state‑of‑the‑art hardware and multi‑modal models.\n\n#J-18808-Ljbffr","company":"Amazon","rawCompany":"amazon","city":"Cupertino","state":"CA","isRemote":false,"isActive":false,"createdAt":"2026-07-16T03:29:39.516Z","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":"541512","title":"Computer Systems Design Services","slug":"computer-systems-design-services"},{"code":"513210","title":"Software Publishers","slug":"software-publishers"},{"code":"518210","title":"Computing Infrastructure Providers, Data Processing, Web Hosting, and Related Services","slug":"computing-infrastructure-providers-data-processing-web-hosting-and-related-services"}],"jobPosting":{"@context":"https://schema.org","@type":"JobPosting","title":"ML Distributed Training Performance Engineer - Trainium","description":"Amazon Web Services (AWS) Annapurna Labs is seeking a software engineer for the Distributed Training team in AWS Neuron to optimize large‑scale ML model training across Trainium. You will work on throughput, memory and compiler/runtime performance across the Neuron stack.\nYou will collaborate with PyTorch and JAX teams and push improvements through the Neuron compiler and runtime, enabling efficient training on state‑of‑the‑art hardware and multi‑modal models.\n\n#J-18808-Ljbffr","datePosted":"2026-07-16T03:29:39.516Z","dateModified":"2026-07-16T03:29:39.516Z","hiringOrganization":{"@type":"Organization","name":"Amazon","sameAs":"https://jobsearcher.com"},"jobLocation":{"@type":"Place","address":{"@type":"PostalAddress","addressLocality":"Cupertino","addressRegion":"CA","addressCountry":"US"}},"identifier":{"@type":"PropertyValue","name":"JobSearcher","value":"42c70cd957e93a42447210a1"},"url":"https://jobsearcher.com/jobs/42c70cd957e93a42447210a1"}}