- 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
- 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
- Meta (Sacramento, 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)
- …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)
- …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 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)
- …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 (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
- DoorDash (San Francisco, CA)
- …logistics engine for delivery! We are bringing on a talented Machine Learning Engineer to help us build the Dasher Acquisition, Mobilization, and Pay systems that ... and Merchants. About the Role As a Machine Learning Engineer , you will work with our robust data and...machine learning and OSS ML technologies such as Spark, PyTorch , Airflow with hands-on experience in production. + Demonstrated… 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
- TEKsystems (San Jose, CA)
- …of at least one mainstream machine learning programming framework (TensorFlow/ PyTorch /MXNet), familiar with its architecture and implementation mechanism. * BS/MS ... relevant connections between users, advertisers, and our client Skills tensorflow, pytorch , machine learning, golang, spark Top Skills Details tensorflow, pytorch… more
- Google (San Francisco, CA)
- …top of industry trends and devise enhancements to Google Cloud products. As a Customer Engineer , you will partner with technical Sales teams as a subject matter ... role. + 7 years of technical sales experience (ie, working as a customer engineer , solutions engineer , or sales engineer ). + Experience engaging with,… 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
- Snap Inc. (Los Angeles, CA)
- …in the way people communicate and express themselves. We are looking for a Research Engineer to join the Creative Vision team. What you'll do: + Contribute to a ... Share your expertise with other teammates and interns + Partner with engineering teams to deliver your technology to...and programming skills + Expert knowledge with Python and PyTorch + Proficiency with python data science stack: NumPy,… 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