- 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 (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
- Google (Seattle, WA)
- …ecosystem at Google by collaborating with some of the resourceful minds ML , compiler design, and systems architecture. Information collected and processed ... Senior Software Engineer, GPU Performance _corporate_fare_ Google _place_...experience in a technical leadership role. + Experience with compiler optimization, code generation, and runtime systems for GPU… more
- Amazon (Seattle, WA)
- …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