• Senior Staff Engineer

    Google (Sunnyvale, CA)
    …compiler infrastructures. + Strong understanding of machine learning frameworks ( JAX , PyTorch or TensorFlow), including their internal architecture and performance ... on and is growing every day. As a software engineer , you will work on a specific project critical...continue to push technology forward. This role within the JAX 3P team involves developing innovative solutions at the… more
    Google (03/13/25)
    - Related Jobs
  • Software Development Engineer - AI/ML , AWS…

    Amazon (Cupertino, CA)
    …a part of the Neuron Frameworks team you'll develop and enhance support PyTorch and JAX for AWS Neuron, working with the open source ecosystem. You will develop and ... the leading ML frameworks, delivering an outstanding user experience for PyTorch and JAX ML model development on the Trainium and Inferentia accelerators. You will… more
    Amazon (01/10/25)
    - Related Jobs
  • Senior Staff Software…

    Google (Sunnyvale, CA)
    …or other similar experience. + Experience with ML frameworks (eg, PyTorch, JAX , TensorFlow). Preferred qualifications: + Master's degree or PhD in Engineering, ... 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's needs with opportunities to… more
    Google (03/14/25)
    - Related Jobs
  • Senior Staff Software…

    Google (Sunnyvale, CA)
    …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 with ... stack through influencing model design, optimizing low-level kernels and compilers (OpenXLA, JAX , Triton), and bridging the gap between model developers and hardware… more
    Google (03/04/25)
    - Related Jobs
  • Senior Software Development Engineer

    Amazon (Seattle, WA)
    …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 is responsible for ... for both latency and throughput on these large models using PyTorch or JAX is a must. Experience with technologies/tools such as vLLM, Hugging Face, multi-modal… more
    Amazon (02/15/25)
    - Related Jobs
  • Sr. Software Engineer - AI/ML, AWS Neuron…

    Amazon (Seattle, WA)
    …Inferentia (Inf1/Inf2) our cloud-scale Machine Learning accelerators. This role is for a Senior Machine Learning Engineer in the Distribute Training team for AWS ... You will lead efforts to build distributed training support into PyTorch and JAX using XLA, the Neuron compiler, and runtime stacks. You will optimize models… more
    Amazon (03/08/25)
    - Related Jobs
  • ML Compiler Engineer , Annapurna Labs

    Amazon (Cupertino, CA)
    …and Vision, originating from leading frameworks such as PyTorch, TensorFlow, and JAX . Your role will involve working closely with our custom-built Machine Learning ... powering solutions like Generative AI. In this role as a ML Compiler engineer , you'll be instrumental in designing, developing, and optimizing features for our… more
    Amazon (01/18/25)
    - Related Jobs
  • Software Engineer - AI/ML, AWS Neuron…

    Amazon (Cupertino, CA)
    …accelerators 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 ... degree in computer science or equivalent - Preferred previous software engineer expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and Frameworks,… more
    Amazon (03/08/25)
    - Related Jobs
  • Sr. Machine Learning Engineer , Amazon…

    Amazon (Seattle, WA)
    …enable breakthrough research and product development across the company. We are seeking a Senior Machine Learning Engineer to join our team and lead the ... for the development and/or acquisition of those skills to senior leaders. You scout top talent and recruit them...understanding of AI frameworks such as PyTorch, TensorFlow, and JAX , and their demands on underlying compute infrastructure, memory… more
    Amazon (02/15/25)
    - Related Jobs
  • Sr. Software Engineer - AI/ML, AWS Neuron…

    Amazon (Cupertino, CA)
    …and Trainium (Trn1), our cloud-scale Machine Learning accelerators. This role is for a senior machine learning engineer in the Distribute Training team for AWS ... we're building an environment that celebrates knowledge-sharing and mentorship. Our senior members enjoy one-on-one mentoring and thorough, but kind, code reviews.… more
    Amazon (03/12/25)
    - Related Jobs