- 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 (Columbus, OH)
- …& 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 (Seattle, WA)
- …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...leap in performance. You: As a Machine Learning Compiler Engineer I on the AWS Neuron Compiler team, you… 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 (Seattle, WA)
- …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 (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
- Staples (Framingham, MA)
- …architecture designs, code development, evaluation, and operationalizing solutions at scale. Partner with a team Software, Data, ML, and Applied engineers for ... at Staples by defining the future roadmap for ongoing work. The AI Engineer is responsible for developing and engineering AI systems and solutions that drive… 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 (Seattle, WA)
- …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 (Bellevue, WA)
- Description We are looking for a Software Development Engineer II (SDEII) to join our Applied AI team to work on LLM-based solutions. You will be responsible for ... required to perform the job in Python, Java, container-based development, SageMaker, PyTorch , PyTorch Lightning, Langchain, etc. Your work should consistently be… more
- Amazon (Herndon, VA)
- …the AWS Cloud. We work together with customer teams and the AWS Partner Network (APN) to execute enterprise cloud computing initiatives. Our team provides assistance ... and productionizing ML models, and using Python and frameworks such as PyTorch , TensorFlow Preferred Qualifications - AWS experience preferred, with proficiency in a… 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
- 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 (Seattle, WA)
- …including Large Language and Vision, originating from leading frameworks such as PyTorch , TensorFlow, and JAX. Your role will involve working closely with our ... Generative AI. Key job responsibilities As a ML Compiler Engineer II on the Neuron Compiler Automated Reasoning Group,...- Experience with deep learning frameworks like TensorFlow or PyTorch - Understanding of large language model (LLM) training… more