• Staff Software Engineer

    Google (Sunnyvale, CA)
    …new problems across the full-stack as we continue to push technology forward. As a Staff Software Engineer on the Core Machine Learning(ML) Frameworks team, ... in building data pipelines for machine learning, including with JAX , TensorFlow, or PyTorch. + Experience with distributed training...goes on and is growing every day. As a software engineer , you will work on a… more
    Google (03/12/25)
    - Save Job - Related Jobs - Block Source
  • Senior Staff Engineer , JAX

    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's needs with ... degree or equivalent practical experience. + 8 years of experience in software development, and with data structures/algorithms. + 5 years of experience leading… more
    Google (03/13/25)
    - Save Job - Related Jobs - Block Source
  • 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 (01/10/25)
    - Save Job - Related Jobs - Block Source
  • Staff Software Engineer

    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 and with data structures/algorithms. + 5 years...to run training and inference workloads using PyTorch and JAX . The machine learning stack team is responsible for… more
    Google (03/04/25)
    - Save Job - Related Jobs - Block Source
  • Sr. Staff Software Engineer

    LinkedIn (Mountain View, CA)
    …using thousands of features), and enabling GPU inference at scale. As a Sr. Staff Software Engineer , you will have first-hand opportunities to advance ... optimize performance across algorithms, AI frameworks, data infra, compute software , and hardware to harness the power of our...billions of user queries. Model Training Infrastructure: As an engineer on the AI Training Infra team, you will… more
    LinkedIn (02/05/25)
    - Save Job - Related Jobs - Block Source
  • Principal Staff Software

    LinkedIn (Mountain View, CA)
    …Online Learning, and advanced LLM Agents work for Training infrastructure. As a Principal Staff Software Engineer on the AI Training Infra team, you ... vision models. We optimize training performance across algorithms, AI frameworks, infrastructure software , and hardware to harness the power of our GPU fleet with… more
    LinkedIn (01/30/25)
    - Save Job - Related Jobs - Block Source
  • 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 (03/15/25)
    - Save Job - Related Jobs - Block Source
  • Senior Staff Software

    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 (03/04/25)
    - Save Job - Related Jobs - Block Source
  • Staff Software Engineer

    Google (Mountain View, 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's needs with ... + 8 years of experience in testing, and launching software products. + 5 years of experience with ...Learning (ML) community to identify performance opportunities and drive TensorFlow/ JAX TPU performance. + Work on scaling numeric and… more
    Google (03/04/25)
    - Save Job - Related Jobs - Block Source
  • Staff Software Engineer , AI…

    Google (Mountain View, 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's needs with ... degree or equivalent practical experience. + 8 years of experience in software development, and with data structures/algorithms. + 5 years of experience with… more
    Google (02/18/25)
    - Save Job - Related Jobs - Block Source
  • Senior Staff Software

    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's needs with ... Bachelor's degree or equivalent practical experience. + 8 years of experience with software development in one or more programming languages (eg, Python, C, C++). +… more
    Google (03/14/25)
    - Save Job - Related Jobs - Block Source
  • 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 (03/08/25)
    - Save Job - Related Jobs - Block Source
  • 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...work safely and cooperatively with other employees, supervisors, and staff ; adhere to standards of excellence despite stressful conditions;… more
    Amazon (03/08/25)
    - Save Job - Related Jobs - Block Source
  • 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)
    - Save Job - Related Jobs - Block Source
  • Software Development Engineer

    Amazon (Cupertino, CA)
    …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. ... Description AWS Neuron is the complete software stack for the AWS Inferentia and Trainium...throughput on these large models using Python, Pytorch or JAX is a must. Deepspeed and other distributed inference… more
    Amazon (02/15/25)
    - Save Job - Related Jobs - Block Source
  • 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)
    - Save Job - Related Jobs - Block Source
  • 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 (01/08/25)
    - Save Job - Related Jobs - Block Source
  • Software Engineer - AI/ML, AWS…

    Amazon (Cupertino, CA)
    …and operations experience - 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...Learning accelerators. This role is for a machine learning engineer in the Distribute Training team for AWS Neuron,… more
    Amazon (03/07/25)
    - Save Job - Related Jobs - Block Source
  • 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)
    - Save Job - Related Jobs - Block Source
  • Sr. Machine Learning Engineer , Amazon…

    Amazon (Sunnyvale, CA)
    …and machine learning initiatives. We leverage advanced hardware, innovative software architectures, and distributed computing techniques to enable breakthrough ... the company. We are seeking a Senior Machine Learning Engineer to join our team and lead the development...A day in the life 8+ years of professional software development experience in distributed systems with emphasis on… more
    Amazon (12/23/24)
    - Save Job - Related Jobs - Block Source