• 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
  • Senior Deep Learning Software

    NVIDIA (Santa Clara, CA)
    We are now looking for a Senior Deep Learning Software Development Engineer ! Academic and commercial groups around the world are using GPUs to power a ... with the world's best talent. Join us in building software which will be used by the entire world...shared with the community. As an experienced and skilled engineer , your collaboration in developing new technology will have… more
    NVIDIA (02/22/25)
    - Save Job - Related Jobs - Block Source
  • Senior Software Engineer , AI…

    LinkedIn (Mountain View, CA)
    …engineers to optimize their models and deliver the best performance possible. As a Senior Software Engineer , you will have first-hand opportunities to ... 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/02/25)
    - Save Job - Related Jobs - Block Source
  • Senior Software Development…

    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
  • 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
  • Senior 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 ... degree or equivalent practice experience. + 5 years of experience with software development in one or more programming languages, and with data… more
    Google (03/04/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
  • Senior Software Engineer

    NVIDIA (Santa Clara, CA)
    …PyTorch open source community to deliver our customers the best of NVIDIA software . In this position you will learn innovative techniques from NVIDIA's domain ... written in Numpy and PyTorch. Beyond building nvFuser influence and improve the entire software stack all the way from users to the CUDA compiler, to the… more
    NVIDIA (12/20/24)
    - Save Job - Related Jobs - Block Source
  • Senior Software Engineer

    NVIDIA (Santa Clara, CA)
    We are looking for an experienced software professional to contribute to design and development of accelerated and distributed implementations of Python APIs for ... libraries (eg NumPy, SciPy), including accelerated implementations (eg CuPy, Jax .NumPy, NumS, cuNumeric) + Experience developing and publishing Python libraries,… more
    NVIDIA (03/05/25)
    - Save Job - Related Jobs - Block Source
  • Sr. Software Engineer - AI/ML, AWS…

    Amazon (Cupertino, CA)
    …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 ... science or equivalent - Experience in computer architecture - Previous software engineering expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and… more
    Amazon (03/08/25)
    - Save Job - Related Jobs - Block Source
  • Senior Deep Learning Compiler…

    NVIDIA (Santa Clara, CA)
    …We build innovative AI compiler solutions that works together with NVIDIA's software stack to provide comprehensive acceleration for modern machine learning models. ... be building a close technical relationship with internal NVIDIA software and hardware teams on push the latest developments...in developing or using deep learning frameworks (eg PyTorch, JAX , TensorFlow, ONNX etc.) + Strong python and C/C++… more
    NVIDIA (03/15/25)
    - Save Job - Related Jobs - Block Source
  • Senior Performance Engineer - Deep…

    NVIDIA (Santa Clara, CA)
    NVIDIA is hiring senior software engineers to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI ... directly contributes to premiere Deep Learning frameworks - PyTorch, JAX and TensorFlow. We work with multiple teams both...advise design of new hardware generations and core platform software components. What we need to see: + BS… more
    NVIDIA (02/02/25)
    - Save Job - Related Jobs - Block Source
  • Software Engineer - AI/ML, AWS…

    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 ... - Bachelor's degree in computer science or equivalent - Preferred previous software engineer expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and… more
    Amazon (03/08/25)
    - Save Job - Related Jobs - Block Source
  • Sr. Software Engineer - AI/ML, AWS…

    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 ... science or equivalent - Experience in computer architecture - Previous software engineering expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and… more
    Amazon (03/12/25)
    - Save Job - Related Jobs - Block Source
  • Senior High-Performance LLM Training…

    NVIDIA (Santa Clara, CA)
    We are now looking for a Senior High-Performance LLM Training Engineer ! NVIDIA is seeking experienced engineers specializing in performance analysis and ... systems. This position focuses on optimizing NVIDIA's high-performance LLM software stack in frameworks like PyTorch and JAX... software stack in frameworks like PyTorch and JAX for high-performance training on thousands of GPUs, while… more
    NVIDIA (03/12/25)
    - Save Job - Related Jobs - Block Source
  • Sr. Software Engineer - AI/ML, AWS…

    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 ... 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
  • 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
  • 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
  • Distinguished Engineer , AI Resiliency Lead

    NVIDIA (Santa Clara, CA)
    We are seeking a Distinguished Engineer to lead AI Resiliency at NVIDIA! Join NVIDIA and help push the boundaries of AI. In this role, you will architect, design, ... and develop world-class software resiliency features for training ground breaking AI models...of frontier models on industry-leading frameworks like PyTorch and JAX /XLA, with near-zero downtime. Your optimizations will span from… more
    NVIDIA (01/22/25)
    - Save Job - Related Jobs - Block Source