• Software Development Engineer

    Amazon (Cupertino, CA)
    Description AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning accelerators. As a part of the Neuron ... Frameworks team you'll develop and enhance support PyTorch and JAX for AWS Neuron, working with the open source...out of the box and delight our customers. Strong software development skills, experience with C++ and Python and… more
    Amazon (03/19/25)
    - Related Jobs
  • Software Engineer - AI/ML, AWS…

    Amazon (Cupertino, CA)
    …- Bachelor's degree in computer science or equivalent - Preferred previous software engineer expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and ... that use them. This role is for a senior software engineer in the Machine Learning Applications...the efforts building distributed training support into Pytorch, Tensorflow, JAX and the Neuron compiler and runtime stacks. This… more
    Amazon (04/01/25)
    - Related Jobs
  • Sr. Software Engineer - AI/ML, AWS…

    Amazon (Cupertino, CA)
    …science or equivalent - Experience in computer architecture - Previous software engineering expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and ... Description Annapurna Labs designs silicon and software that accelerates innovation. Customers choose us to...accelerators. This role is for a Senior Machine Learning Engineer in the Distribute Training team for AWS Neuron,… more
    Amazon (04/02/25)
    - Related Jobs
  • Senior Software Development Engineer

    Amazon (Cupertino, CA)
    …accelerators and the Trn1 and Inf2 servers that use them. This role is for a software engineer in the Neuron Inference (ML Apps) team for AWS Neuron. This role ... Description AWS Neuron is the complete software stack for the AWS Inferentia and Trainium...and throughput on these large models using PyTorch or JAX is a must. Experience with technologies/tools such as… more
    Amazon (04/24/25)
    - Related Jobs
  • Sr. Software Engineer - AI/ML, AWS…

    Amazon (Cupertino, CA)
    …science or equivalent - Experience in computer architecture - Previous software engineering expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and ... Labs (our organization within AWS UC) designs silicon and software that accelerates innovation. Customers choose us to create...accelerators. This role is for a senior machine learning engineer in the Distribute Training team for AWS Neuron,… more
    Amazon (03/12/25)
    - Related Jobs
  • Senior Software Engineer , GPU…

    Google (Sunnyvale, CA)
    …UI design and mobile; the list goes on and is growing every day. As a software engineer , you will work on a specific project critical to Google Cloud's needs ... degree or equivalent practice experience. + 5 years of experience with software development in one or more programming languages, and with data… more
    Google (05/04/25)
    - Related Jobs
  • Sr. Software Engineer - AI/ML, AWS…

    Amazon (Cupertino, CA)
    …and the Trn1 and Inf1 servers that use them. This role is for a senior software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. This ... services. Annapurna Labs (our organization within AWS UC) designs silicon and software that accelerates innovation. Customers choose us to create cloud solutions… more
    Amazon (04/18/25)
    - Related Jobs
  • Staff Software Engineer , GPU…

    Google (Sunnyvale, CA)
    …UI design and mobile; the list goes on and is growing every day. As a software engineer , you will work on a specific project critical to Google Cloud's needs ... equivalent practical experience. + 8 years of experience in software development. + 5 years of experience testing, and...model design, optimizing low-level kernels and compilers (eg, OpenXLA, JAX , Triton, etc.), and bridging the gap between model… more
    Google (04/17/25)
    - Related Jobs
  • Senior Staff Software Engineer , GPU…

    Google (Sunnyvale, CA)
    …UI design and mobile; the list goes on and is growing every day. As a software engineer , you will work on a specific project critical to Google Cloud's needs ... Bachelor's degree or equivalent practical experience. + 8 years of experience in software development. + 7 years of experience leading technical project strategy, ML… more
    Google (04/04/25)
    - Related Jobs
  • Sr. Software Engineer - AI/ML, AWS…

    Amazon (Cupertino, CA)
    …and the Trn1 and Inf1 servers that use them. This role is for a senior software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. This ... portfolio. Annapurna Labs (our organization within AWS UC) designs silicon and software that accelerates innovation. Customers choose us to create cloud solutions… more
    Amazon (02/09/25)
    - Related Jobs