Job ID: 3047146 | Amazon.com Services LLC About the Product: AWS Machine Learning accelerators are leading innovation at AWS, essential for building Generative AI. Our Inferentia chip offers unparalleled ML inference performance, while Trainium provides exceptional ML training capabilities, delivering optimal performance with impressive teraflops (TFLOPS) of compute power. The AWS Neuron Software Development Kit (SDK) includes a state-of-the-art ML compiler and runtime, seamlessly integrating with popular ML frameworks like PyTorch, TensorFlow, and MXNet. Our solutions serve significant clients, including Snap, Autodesk, Amazon Alexa, and Amazon Rekognition. About the Team: The Annapurna Labs team at Amazon is responsible for silicon development, covering a wide range of disciplines such as silicon engineering, hardware design, software, and operations. The AWS Neuron team focuses on optimizing complex neural net models for our custom-made AWS hardware. We are diligently developing a deep learning compiler stack that efficiently translates neural network models from frameworks like TensorFlow, PyTorch, and MXNet into executable code. Our team comprises industry-leading experts dedicated to achieving groundbreaking performance improvements. About You: As a Senior Machine Learning Compiler Engineer, you will play a pivotal role in the design and development of our compiler, catering to the world’s largest ML workloads. You will architect and implement critical features, share innovative research, and mentor a talented team of engineers. Effective technical communication will be vital as you collaborate with AWS ML services teams. Your involvement will encompass pre-silicon design and the introduction of new products/features. While a background in Machine Learning and AI accelerators is preferred, it is not mandatory. Location: Candidates should be located, or willing to relocate, to Cupertino (preferred), Seattle, Austin, or Toronto. Inclusive Team Culture: At AWS, we celebrate diversity and are committed to fostering an inclusive culture. With ten employee-led affinity groups, we strive to ensure each voice is heard. We offer innovative benefits and host annual learning experiences focused on diversity. Work-Life Balance: We prioritize a healthy work-life balance. Your success is measured not by hours spent, but by the energy you bring to both your personal and professional life. Flexibility in working hours is encouraged to achieve this balance. Mentorship & Career Growth: Our team values knowledge sharing and mentorship. We celebrate diverse experience levels and aim to support your career growth by assigning projects tailored to enhance your skills. BASIC QUALIFICATIONS: 5+ years of professional software development experience (non-internship) 5+ years of programming experience with at least one software language 5+ years of experience leading design or architecture of systems 5+ years of experience in the complete software development life cycle Experience as a mentor, tech lead, or leading an engineering team PREFERRED QUALIFICATIONS: Bachelor's degree in computer science or equivalent This position requires compliance with company policies and standards, including adherence to a safe and respectful work environment. A diverse workforce is essential for Amazon’s success. We welcome applications from all qualified candidates, including those with disabilities.