Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

2 giorni fa

Turin, Italia Vendita al dettaglio e all'ingrosso Import-export A tempo pieno

Sr. Software Engineer- AI/ML, AWS Neuron Distributed TrainingAnnapurna Labs designs silicon and software that accelerate innovation. Customers choose us to create cloud solutions that solve challenges that were unimaginable a short time ago—even yesterday. Our custom chips, accelerators, and software stacks enable us to take on technical challenges that have never been seen before, and deliver results that help our customers change the world.AWS Neuron is the complete software stack for the AWS Trainium (Trn1/Trn2) and Inferentia (Inf1/Inf2) cloud‑scale Machine Learning accelerators. This role is for a Senior Machine Learning Engineer in the Distributed Training team for AWS Neuron, responsible for development, enablement and performance tuning of a wide variety of ML model families, including massive‑scale Large Language Models (LLM) such as GPT and Llama, as well as Stable Diffusion, Vision Transformers (ViT) and many more.Key ResponsibilitiesLead efforts to build distributed training support into PyTorch and JAX using XLA, the Neuron compiler, and runtime stacks.Optimize models to achieve peak performance and maximize efficiency on AWS custom silicon, including Trainium and Inferentia, as well as Trn2, Trn1, Inf1, and Inf2 servers.Work with cross‑functional teams (chip architects, compiler engineers, runtime engineers) to create, build and tune distributed training solutions with Trainium instances.Deep dive into model training pipelines and extend distributed training libraries such as FSDP, Deepspeed, Nemo, and others for the Neuron system.About the TeamOur team collaborates closely with hardware and software experts to support new members, promote knowledge sharing, and ensure high‑quality code through mentoring and thorough code reviews. We value diversity and inclusive culture, providing flexibility, mentorship, and growth opportunities for all team members.Basic QualificationsBachelor’s degree in computer science or equivalent.5+ years of noninternship professional software development experience.5+ years of programming in at least one software programming language.5+ years of leading design or architecture of new and existing systems.5+ years of full software development life cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.Experience as a mentor, tech lead or engineering team lead.Experience in machine learning, data mining, information retrieval, statistics or natural language processing.Preferred QualificationsMaster’s degree in computer science or equivalent.Experience in computer architecture.Previous software engineering expertise with PyTorch/JAX/TensorFlow, distributed libraries and frameworks, and end‑to‑end model training.Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.Location: US, Massachusetts, North Reading. Salary: $193,300 - $261,500 annually. Benefits include health insurance, 401(k) matching, paid time off, parental leave, and more. View full benefits at

  • ML Systems Engineer: Distributed Training on Neuron/Trainium

    2 settimane fa

    Turin, Italia Amazon A tempo pieno

    A leading technology company in Torino is seeking a Software Engineer to join the Machine Learning Applications team for AWS Neuron. This role will focus on developing and tuning distributed training solutions for large models, including PyTorch and JAX support. The ideal candidate will have extensive experience in software development, machine learning, and...

  • Software Engineer- AI/ML, AWS Neuron

    2 settimane fa

    Turin, Italia Amazon A tempo pieno

    AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud‑scale machine learning accelerators and the Trn1 and Inf1 servers that use them. This role is for a software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. This role is responsible for development, enablement and performance tuning of a wide...

  • ML Kernel Performance Engineer, AWS Neuron, Annapurna Labs

    2 settimane fa

    Turin, Italia Amazon A tempo pieno

    ML Kernel Performance Engineer, AWS Neuron, Annapurna Labs The 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 is at the forefront of maximizing...

  • EFA Engineer, AWS Annapurna Labs

    2 settimane fa

    Turin, Italia Amazon A tempo pieno

    Join Annapurna Labs and take part in the development of next generation products that will define the future of the cloud. Annapurna Labs, an Amazon company, is looking for a senior software engineer with experience in networking and software to join the networking SW group. The ideal candidate thrives in complex software and hardware designs, takes...

  • DataOps & MLOps Engineer

    2 settimane fa

    Turin, Italia ALLSIDES A tempo pieno

    About ALLSIDES ALLSIDES is redefining how the world experiences 3D content. We combine physically accurate scanning and generative AI to power content creation workflows for e-commerce, virtual environments, and immersive experiences. Our clients include global brands like adidas, Meta, Amazon, and Zalando. We operate a rapidly scaling photorealistic 3D...

  • Senior Staff Engineer

    5 giorni fa

    Turin, Italia Turing A tempo pieno

    About Turing:Based in San Francisco, California, Turing is the world's leading research accelerator for frontier AI labs and a trusted partner for global enterprises deploying advanced AI systems. Turing supports customers in two ways: first, by accelerating frontier research with high-quality data, advanced training pipelines, plus top AI researchers who...

  • Senior Staff Engineer

    4 settimane fa

    Turin, Italia Turing A tempo pieno

    About Turing: Based in San Francisco, California, Turing is the world's leading research accelerator for frontier AI labs and a trusted partner for global enterprises deploying advanced AI systems. Turing supports customers in two ways: first, by accelerating frontier research with high-quality data, advanced training pipelines, plus top AI researchers who...

  • Senior Staff Engineer

    1 settimana fa

    Turin, Italia Turing A tempo pieno

    About Turing: Based in San Francisco, California, Turing is the world's leading research accelerator for frontier AI labs and a trusted partner for global enterprises deploying advanced AI systems. Turing supports customers in two ways: first, by accelerating frontier research with high-quality data, advanced training pipelines, plus top AI researchers who...

  • Senior Staff Engineer

    5 giorni fa

    Turin, Italia Turing A tempo pieno

    About Turing: Based in San Francisco, California, Turing is the world's leading research accelerator for frontier AI labs and a trusted partner for global enterprises deploying advanced AI systems. Turing supports customers in two ways: first, by accelerating frontier research with high-quality data, advanced training pipelines, plus top AI researchers who...

  • Sr. Systems Engineer, Managed Operations

    2 settimane fa

    Turin, Italia Amazon A tempo pieno

    Sr. Systems Engineer, Managed Operations Job ID: | Amazon Spain Services, S.L.U. AWS is set to introduce the inaugural European Sovereign Cloud (ESC), marking a significant development in utility computing (UC). To spearhead this initiative, we are actively seeking experienced senior systems engineers with a strong background in automation and operations. As...