- Amazon (Cupertino, CA)
- …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 ecosystem. You will ... leading ML frameworks, delivering an outstanding user experience for PyTorch and JAX ML model development on the Trainium...you're applying in isn't listed, please contact your Recruiting Partner . Our compensation reflects the cost of labor across… more
- Amazon (Cupertino, CA)
- …(Inf1/Inf2) our cloud-scale Machine Learning accelerators. This role is for a Machine Learning Engineer on one of our AWS Neuron teams: - The ML Distributed Training ... make AWS Trainium and Inferentia feel native inside the tools builders already love- PyTorch , JAX, and the rapidly evolving vLLM ecosystem. By weaving Neuron SDK deep… more
- Amazon (Cupertino, CA)
- …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 ecosystem. You will ... leading ML frameworks, delivering an outstanding user experience for PyTorch and JAX ML model development on the Trainium...you're applying in isn't listed, please contact your Recruiting Partner . Our compensation reflects the cost of labor across… more
- Amazon (Cupertino, CA)
- …an ML compiler, runtime and natively integrates into popular ML frameworks, such as PyTorch , TensorFlow and MxNet. AWS Neuron and Inferentia are used at scale with ... neural network descriptions created in frameworks such as TensorFlow, PyTorch , and MXNET, and converts them into code suitable...a quantum leap in performance. You: Machine Learning Compiler Engineer II on the AWS Neuron team, you will… more
- Amazon (Cupertino, CA)
- …and application framework, which seamlessly integrate into popular ML frameworks like PyTorch . AWS Neuron, running on Inferentia and Trainium, is trusted and used ... and Vision models created in frameworks such as TensorFlow, PyTorch , and JAX, and makes them run performantly on...in performance. You: As a PhD Machine Learning Compiler Engineer on the AWS Neuron Compiler team, you will… more
- Amazon (Cupertino, CA)
- …an ML compiler, runtime and natively integrates into popular ML frameworks, such as PyTorch , TensorFlow and JAX. AWS Neuron is used at scale with customers like ... neural network descriptions created in frameworks such as TensorFlow, PyTorch , and JAX, and converts them into code suitable...quantum leap in performance. As a Machine Learning Compiler Engineer II in the AWS Neuron Compiler team, you… more
- Amazon (Santa Clara, CA)
- …you will be responsible for contributing extensions to the TensorFlow and PyTorch machine learning frameworks and for developing cross-framework solutions to support ... the space. Key job responsibilities As a Software Development Engineer in the SageMaker Engines team, you will be...as pipeline and tensor parallelism as extensions to the PyTorch framework; - Implementing sharding of the model training… more
- Amazon (Santa Clara, CA)
- …you will be responsible for contributing extensions to the TensorFlow and PyTorch machine learning frameworks and for developing cross-framework solutions to support ... cloud services. Key job responsibilities As a Software Development Engineer in the SageMaker Engines team, you will be...as pipeline and tensor parallelism as extensions to the PyTorch framework; - Implementing sharding of the model training… more
- Amazon (Cupertino, CA)
- …Machine Learning accelerators. This role is for a Senior Machine Learning Engineer in the Distribute Training team for AWS Neuron, responsible for development, ... develop and improve distributed training capabilities in popular machine learning frameworks ( PyTorch and JAX) using AWS's specialized AI hardware. Working with our… more
- Amazon (Cupertino, CA)
- …degree in computer science or equivalent - Preferred previous software engineer expertise with Pytorch /Jax/Tensorflow, Distributed libraries and Frameworks, ... Web Services (AWS) is looking for a Software Development Engineer II to build, deliver, and maintain complex products...help lead the efforts building distributed training support into Pytorch , Tensorflow using XLA and the Neuron compiler and… more
- Amazon (Cupertino, CA)
- …our custom chips designed to accelerate deep-learning workloads. This role is for a software engineer in the Compiler team for AWS Neuron. As part of this role, you ... generation Neuron compiler which transforms ML models written in ML frameworks (eg, PyTorch , TensorFlow, and JAX) to be deployed AWS Inferentia and Trainium based… more
- Amazon (Cupertino, CA)
- …an ML compiler, runtime and natively integrates into popular ML frameworks, such as PyTorch , TensorFlow and MxNet. AWS Neuron and Inferentia are used at scale with ... neural network descriptions created in frameworks such as TensorFlow, PyTorch , and MXNET, and converts them into code suitable...in performance. You: As a Sr. Machine Learning Compiler Engineer III on the AWS Neuron team, you will… more
- Amazon (Cupertino, CA)
- …an ML compiler, runtime and natively integrates into popular ML frameworks, such as PyTorch , TensorFlow and MxNet. AWS Neuron and Inferentia are used at scale with ... neural network descriptions created in frameworks such as TensorFlow, PyTorch , and MXNET, and converts them into code suitable...in performance. You: As a Sr. Machine Learning Compiler Engineer III on the AWS Neuron team, you will… more
- NVIDIA (Santa Clara, CA)
- …and Agentic AI workflows. The NeMo Retriever team is looking for an AI Engineer to join our team, focusing on the intersection of machine learning development, ... using Python or an equivalent performant language. + Work closely with partner teams to understand requirements, build & evaluate POCs, and develop roadmaps… more
- Amazon (Cupertino, CA)
- …including Large Language and Vision, originating from leading frameworks such as PyTorch , TensorFlow, and JAX. Your role will involve working closely with our ... AI. Key job responsibilities As a Sr. ML Compiler Engineer III on the Neuron Compiler Automated Reasoning Group,...- Experience with deep learning frameworks like TensorFlow or PyTorch - Understanding of large language model (LLM) training… more
- Amazon (Cupertino, CA)
- …Machine Learning accelerators. This role is for a Senior Machine Learning Engineer in the Distribute Training team for AWS Neuron, responsible for development, ... will lead efforts to build distributed training support into PyTorch and JAX using XLA, the Neuron compiler, and...you're applying in isn't listed, please contact your Recruiting Partner . Our compensation reflects the cost of labor across… more
- Amazon (Cupertino, CA)
- …Machine Learning accelerators. This role is for a senior machine learning engineer in the Distribute Training team for AWS Neuron, responsible for development, ... You will help lead the efforts building distributed training support into Pytorch , Tensorflow using XLA and the Neuron compiler and runtime stacks. You… more
- Meta (Menlo Park, CA)
- …deliver a great experience for our customers and users.As a Business Support Engineer , you will understand industry trends, our partner 's network evolution ... our business AI platform. We are looking for an engineer to play a key role in providing technical...that run our business AI infrastructure as well as partner specific integrations that utilize our platform 5. Analyze… more
- Amazon (Cupertino, CA)
- …disciplines Software and Hardware engineers including but not limited to complier engineer , machine learning engineer , runtime engineer , performance ... TVM, MLIR, LLVM - Deep learning models and algorithms - Tensorflow, PyTorch , or MxNET frameworks Preferred Qualifications - Knowledge of code generation, compute… more
- Google (Mountain View, CA)
- Snapshot We are seeking a Forward Deployment Engineer , Applied AI to lead the development and deployment of novel applications, leveraging Google's generative AI ... focuses on rapidly developing new features, and working across partner teams to deliver solutions, and maximize impact for...set you up for success as a Forward Deployment Engineer at Google DeepMind, we look for the following… more