We interpreted Mountain View, CA as Mountain View, CA. Other options include: Mountain View (Contra Costa County), CA
- Meta (Menlo Park, CA)
- **Summary:** Meta is seeking a PyTorch Partner Engineer to join Meta's AI Partner Engineering team. This highly technical team collaborates with ... ensuring they are optimized and scalable for real-world applications. **Required Skills:** Partner Engineer , PyTorch Responsibilities: 1. Collaborate with… more
- NVIDIA (Santa Clara, CA)
- …and NVSHMEM for Deep Learning and HPC applications. We are looking for a motivated Partner Enablement Engineer to guide our key partners and customers with NCCL. ... of Machine Learning concepts and experience with Deep Learning Frameworks such PyTorch , TensorFlow + Deep understanding of technology and passionate about what you… 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
- Meta (Sunnyvale, CA)
- …& Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a ... working on one of the core areas such as PyTorch framework components, AI compiler and runtime, high-performance kernels...lower them efficiently on MTIA hardware. You will also partner with hardware design teams to develop compiler optimizations… 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 (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)
- …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 Sr. Machine Learning Compiler Engineer III on 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 ... 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)
- …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)
- …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)
- …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)
- …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 ... both latency and throughput on these large models using PyTorch or JAX is a must. Experience with technologies/tools...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, ... 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
- 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 (Santa Clara, CA)
- …the world's customers we serve! Key job responsibilities As a Software Development Engineer in the SageMaker team, you will be responsible for: - Developing ... companies around the world and open source community - PyTorch , NVIDIA/GPU * Create innovative products to run at...you're applying in isn't listed, please contact your Recruiting Partner . Our compensation reflects the cost of labor across… more
- Capital One (San Jose, CA)
- AI Engineer **Overview:** At Capital One, we are creating responsible and reliable AI systems, changing banking for good. For years, Capital One has been an industry ... for the highest leverage impact. **In this role, you will:** + Partner with a cross-functional team of engineers, research scientists, technical program managers,… 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