- Amazon (Sunnyvale, CA)
- …as well as productizing it on consumer devices, while supporting both current and future compiler and ML accelerator architectures. In this role you will lead a ... team working alongside partner science teams to develop the compiler infrastructure and lower deep learning workloads to heterogeneous device backends. You will also… more
- Amazon (Seattle, WA)
- …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 (Seattle, WA)
- …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 ... team is at the forefront of maximizing performance for AWS's custom ML accelerators. Working at the hardware-software boundary, our engineers craft high-performance… more
- Microsoft Corporation (Mountain View, CA)
- …OneDrive, Skype, Teams and Xbox Live. We are looking for a Principal AI Architect to join our team! **Responsibilities** + Model Bring-Up & Characterization + Lead ... Hardware/Software/Model Co-Design + Partner with silicon and system architects, compiler /runtime engineers, and model researchers to define co-design strategies that… more
- Amazon (Seattle, WA)
- …Neuron As a Technical Leader at the forefront of AWS's AI Accelerator, you'll architect the bridge between ML frameworks including PyTorch, JAX and AI hardware. ... stack powering AWS's next-generation AI accelerators Inferentia and Trainium. As a Senior Software Engineer in our Machine Learning Applications team, you'll be at… more
- NVIDIA (Santa Clara, CA)
- …inference systems that serve large-scale models with extreme efficiency. You'll architect and implement high-performance inference stacks, optimize GPU kernels and ... across multi-GPU, multi-node, and multi-cloud environments. You'll collaborate across inference, compiler , scheduling, and performance teams to push the frontier of… more
- Amazon (Austin, TX)
- …parameters? Build distributed systems that scale across thousands of accelerators? Create compiler optimizations that make ML training blazingly fast? Whether ... your passion lies in hardware design, distributed systems, compiler development, or ML infrastructure, we'll help...AI innovation. What sets us apart? Many of our senior engineers began their careers as interns here, creating… more
- NVIDIA (Austin, TX)
- …in modern high-performance workloads and models. We are looking for an outstanding Senior Software Engineer that can architect and implement these highly ... of the team, you will develop new innovative workflows, work on compiler - or runtime-driven solutions that accelerate critical workloads, generate optimal code… more