Machine Learning Compiler Engineer, Annapurna Labs

The AWS Neuron Compiler team is actively seeking skilled compiler engineers to join our efforts in developing a state-of-the-art deep learning compiler stack. This stack is designed to optimize application models across diverse domains, including Large Language and Vision, originating from leading frameworks such as PyTorch, TensorFlow, and JAX. Your role will involve working closely with our custom-built Machine Learning accelerators, including Inferentia/Trainium, which represent the forefront of AWS innovation for advanced ML capabilities, powering solutions like Generative AI. As a ML Compiler engineer, you will design, develop, and optimize features for our compiler, tackling crucial challenges alongside a talented engineering team. You will collaborate closely with cross-functional team members from the Runtime, Frameworks, and Hardware teams to ensure system-wide performance optimization. As part of the Backend team, you will design and develop various aspects of our system, including instruction scheduling, memory allocation, data transfer optimization, graph partitioning, parallel programming, code generation, Instruction Set Architectures, new hardware bring-up, and hardware-software co-design. Key Responsibilities Solve challenging technical problems, often ones not solved before, at every layer of the stack. Design, implement, test, deploy, and maintain innovative software solutions to transform service performance, durability, cost, and security. Research implementations that deliver the best possible experiences for customers. A Day in the Life Build high-impact solutions to deliver to our large customer base. Participate in design discussions, code review, and communicate with internal and external stakeholders. Work cross-functionally to help drive business decisions with your technical input. Work in a startup-like development environment, where you’re always working on the most important stuff. Basic Qualifications B.S. or M.S. in computer science or related field Proficiency with 1 or more of the following programming languages: C++ (preferred), Python 3 years of non-internship professional software development experience 2 years of experience developing compiler optimization, graph-theory, hardware bring-up, FPGA placement and routing algorithms, or hardware resource management Preferred Qualifications M.S. or Ph.D. in computer science or related field Strong knowledge in one or more of the areas of: compiler design, instruction scheduling, memory allocation, data transfer optimization, graph partitioning, parallel programing, code generation, Instruction Set Architectures, new hardware bring-up, and hardware-software co-design Experience with LLVM and/or MLIR Experience with developing algorithms for simulation tools Experience with TensorFlow, PyTorch, and/or JAX Experience in LLM, Vision or other deep-learning models Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status. Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. This position will remain posted until filled. Applicants should apply via our internal or external career site. J-18808-Ljbffr

Location: Asti, Piemonte, IT

Posted Date: 10/20/2025