- 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...future. Basic Qualifications - 5+ years of non-internship professional software development experience - 5+ years of programming with… more
- LinkedIn (Mountain View, CA)
- …engineers to optimize their models and deliver the best performance possible. As a 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
- 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 ... + Experience working with deep learning frameworks (eg, TensorFlow, PyTorch, JAX ). Google's software engineers develop the next-generation technologies that… more
- LinkedIn (Mountain View, CA)
- …Learning, Online Learning, and advanced LLM Agents work for Training infrastructure. As a Senior 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
- Amazon (Cupertino, CA)
- …and operations experience - Experience in computer architecture - Previous software engineering expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and ... Description AWS Neuron is the complete software stack for the AWS Inferentia (Inf1/Inf2) and...Learning accelerators. This role is for a machine learning engineer in the Inference team for AWS Neuron, responsible… more
- 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 (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 (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 (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 (Palo Alto, CA)
- …information retrieval, and large-scale distributed systems. Key job responsibilities As a Software Development Engineer in Visual Search you will: * Design, ... collaborative environment. Basic Qualifications - 3+ years of non-internship professional software development experience - 3+ years of non-internship design or… more
- Amazon (Cupertino, CA)
- …of compute power for ML in the cloud. This is all enabled by cutting edge software stack, the AWS Neuron Software Development Kit (SDK), which includes an ML ... into popular ML frameworks, such as PyTorch, TensorFlow and JAX . AWS Neuron is used at scale with customers...leap in performance. You: As a Machine Learning Compiler Engineer II in the AWS Neuron Compiler team, you… more
- 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 (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 ... credible, and curious in your own right as a trusted AWS Neuron engineer , innovating on behalf of our customers. Your responsibilities will involve tackling crucial… more
- 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 ... credible, and curious in your own right as a trusted AWS Neuron engineer , innovating on behalf of our customers. Your responsibilities will involve tackling crucial… more