- Google (Sunnyvale, CA)
- …compiler infrastructures. + Strong understanding of machine learning frameworks ( JAX , PyTorch or TensorFlow), including their internal architecture and performance ... on and is growing every day. As a software engineer , you will work on a specific project critical...continue to push technology forward. This role within the JAX 3P team involves developing innovative solutions at the… more
- Amazon (Cupertino, CA)
- …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 develop and ... the leading ML frameworks, delivering an outstanding user experience for PyTorch and JAX ML model development on the Trainium and Inferentia accelerators. You will… more
- Google (Sunnyvale, CA)
- …or other similar experience. + Experience with ML frameworks (eg, PyTorch, JAX , TensorFlow). Preferred qualifications: + Master's degree or PhD in Engineering, ... 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 opportunities to… more
- Google (Sunnyvale, CA)
- …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 Cloud's needs with ... stack through influencing model design, optimizing low-level kernels and compilers (OpenXLA, JAX , Triton), and bridging the gap between model developers and hardware… more
- Amazon (Seattle, WA)
- …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 ... for both latency and throughput on these large models using PyTorch or JAX is a must. Experience with technologies/tools such as vLLM, Hugging Face, multi-modal… more
- Amazon (Seattle, WA)
- …Inferentia (Inf1/Inf2) our cloud-scale Machine Learning accelerators. This role is for a Senior Machine Learning Engineer in the Distribute Training team for AWS ... You will lead efforts to build distributed training support into PyTorch and JAX using XLA, the Neuron compiler, and runtime stacks. You will optimize models… 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)
- …accelerators and the Trn1 and Inf1 servers that use them. This role is for a senior software engineer in the Machine Learning Applications (ML Apps) team for AWS ... degree in computer science or equivalent - Preferred previous software engineer expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and Frameworks,… more
- Amazon (Seattle, WA)
- …enable breakthrough research and product development across the company. We are seeking a Senior Machine Learning Engineer to join our team and lead the ... for the development and/or acquisition of those skills to senior leaders. You scout top talent and recruit them...understanding of AI frameworks such as PyTorch, TensorFlow, and JAX , and their demands on underlying compute infrastructure, memory… more
- Amazon (Cupertino, CA)
- …and Trainium (Trn1), our cloud-scale Machine Learning accelerators. This role is for a senior machine learning engineer in the Distribute Training team for AWS ... we're building an environment that celebrates knowledge-sharing and mentorship. Our senior members enjoy one-on-one mentoring and thorough, but kind, code reviews.… more