- Amazon (Cupertino, CA)
- …and Trainium ML accelerators. This comprehensive toolkit includes an ML compiler , runtime, and application framework that seamlessly integrates with ... popular ML frameworks like PyTorch and JAX enabling unparalleled ...layers - from frameworks and kernels and collaborate with compiler to runtime and collectives. We not only optimize… more
- Amazon (Cupertino, CA)
- …and Trainium ML accelerators. This comprehensive toolkit includes an ML compiler , runtime, and application framework that seamlessly integrates with ... popular ML frameworks like PyTorch and JAX enabling unparalleled ...layers - from frameworks and kernels and collaborate with compiler to runtime and collectives. We not only optimize… more
- Amazon (Cupertino, CA)
- …and it's all being enabled by AWS Neuron. Neuron is a Software that include ML compiler and native integration into popular ML frameworks. Our products ... in providing profiling and optimization tools for the Neuron ML accelerators fleet. You will work closely with the...Tools team * Work closely with the frameworks and compiler teams. * Collect requirements from various other teams… more
- Amazon (Cupertino, CA)
- …of emergent technologies. We're looking for an ASIC Physical Design Methodology Engineer to help us trail-blaze new technologies and architectures, while ensuring ... and deploy innovative physical design and verification methodologies (RTL2GDS) for ML Accelerator chips in advanced nodes Drive Optimizations in CAD… more
- Amazon (Cupertino, CA)
- …learning accelerators and servers that use them. This role is for a software engineer in the Machine Learning Inference Model Enablement team for AWS Neuron at ... Inference Model Enablement team works side by side with compiler engineers and runtime engineers to create, build and...silicon and servers. Strong software development using Python and ML knowledge are both critical to this role. A… more
- Amazon (Cupertino, CA)
- …and Inf1 servers that use them. This role is for a senior software engineer in the Machine Learning Applications ( ML Apps) team for AWS Neuron. This role ... Web Services (AWS) is looking for a Software Development Engineer II to build, deliver, and maintain complex products...and F1 EC2 Instances, AWS Neuron, Inferentia and Trainium ML Accelerators, and in storage with scalable NVMe, are… more
- Microsoft Corporation (Mountain View, CA)
- …software that is accessible and impactful. We're looking for a talented and team-oriented Senior Software Engineer to help shape the future of AI compilation. ... Profile AI kernels and models to uncover opportunities for algorithmic and compiler -level improvements. + Collaborate with hardware architects, ML scientists,… more
- Amazon (Cupertino, CA)
- …edge software stack, the AWS Neuron Software Development Kit (SDK), which includes an ML compiler , Neuron Kernel Interface (NKI) compiler , and runtime that ... - Bachelor's degree in computer science or equivalent - Experience in compiler design for CPU/GPU/Vector engines/ ML -accelerators Amazon is an equal opportunity… more
- NVIDIA (Santa Clara, CA)
- …across multi-GPU, multi-node, and multi-cloud environments. You'll collaborate across inference, compiler , scheduling, and performance teams to push the frontier of ... + Develop, optimize, and benchmark GPU kernels (hand-tuned and compiler -generated) using techniques such as fusion, autotuning, and memory/layout optimization;… more
- Google (Mountain View, CA)
- …modern ML framework architectures (JAX, TensorFlow, PyTorch) with the knowledge of ML compiler toolchains (eg, XLA). + Experience with performance I/O, data ... Senior Staff Software Engineer , Machine Learning...+ 7 years of experience leading technical project strategy, ML design, and working with ML infrastructure… more