Machine Learning Compiler Engineer, Annapurna Labs
نُشرت في :
04-08-2025
صاحب العمل نشط
1 وظيفة شاغرة
حالة تأهب وظيفة
سيتم تحديثك بأحدث تنبيهات الوظائف عبر البريد الإلكترونيValid email field required
أرسل الوظائف
شارك هذه الوظيفة
أرسل لي وظائف مشابهة
حالة تأهب وظيفة
سيتم تحديثك بأحدث تنبيهات الوظائف عبر البريد الإلكترونيValid email field required
أرسل الوظائف
الوصف الوظيفي
Our team is responsible for the AWS Neuron Compiler a cutting-edge deep learning compiler stack that powers Generative AI and other advanced ML workloads on AWSs custom-built ML accelerators Inferentia and Trainium. These accelerators deliver best-in-class performance and cost-efficiency for ML inference and training in the cloud.
Were building a new core group of engineers in TLV to drive innovation in compiler technology and systems-level ML software. As a Machine Learning Compiler Engineer youll help shape the direction of the team from the ground up and work on:
Designing and optimizing compiler features for high-performance ML workloads
Solving complex challenges in areas like graph optimization code generation and hardware-software co-design
Collaborating across Runtime Frameworks and Hardware teams to maximize end-to-end performance
As part of the Backend Compiler Team youll contribute to projects involving instruction scheduling memory management parallelism graph partitioning ISA design and new hardware bring-up.
This is a unique opportunity to be at the intersection of ML and systems within AWS helping to build the future of AI infrastructure right here in Tel Aviv.
Key job responsibilities
Our engineers collaborate across diverse teams projects and environments to have a firsthand impact on our global customer base. You will:
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
As you design and code solutions to help our team drive efficiencies in software architecture youll create metrics implement automation and other improvements and resolve the root cause of software defects. Youll also:
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 youre always working on the most important stuff.
Considering the technical requirements of this role and to ensure fair and equal candidate consideration we are conducting full interview loops. If you meet the basic qualifications and the team is interested in moving forward we will be in touch to coordinate your interviews.
About the team
*Diverse Experiences
Amazon values diverse experiences. Even if you do not meet all of the preferred qualifications and skills listed in the job description we encourage candidates to apply. If your career is just starting hasnt followed a traditional path or includes alternative experiences dont let it stop you from applying.
*Why AWS
Amazon Web Services (AWS) is the worlds most comprehensive and broadly adopted cloud platform. We pioneered cloud computing and never stopped innovating thats why customers from the most successful startups to Global 500 companies trust our robust suite of products and services to power their businesses.
*Work/Life Balance
We value work-life harmony. Achieving success at work should never come at the expense of sacrifices at home which is why flexible work hours and arrangements are part of our culture. When we feel supported in the workplace and at home theres nothing we cant achieve in the cloud.
*Inclusive Team Culture
AWS values curiosity and connection. Our employee-led and company-sponsored affinity groups promote inclusion and empower our people to take pride in what makes us unique. Our inclusion events foster stronger more collaborative teams. Our continual innovation is fueled by the bold ideas fresh perspectives and passionate voices our teams bring to everything we do.
*Mentorship and Career Growth
Were continuously raising our performance bar as we strive to become Earths Best Employer. Thats why youll find endless knowledge-sharing mentorship and other career-advancing resources here to help you develop into a better-rounded professional.
- 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
- 3 years of experience developing compiler optimization graph-theory hardware bring-up FPGA placement and routing algorithms or hardware resource management
- 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 is TensorFlow PyTorch and/or JAX
- Experience in LLM Vision or other deep-learning models
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 including support for the interview or onboarding process please visit for more information. If the country/region youre applying in isnt listed please contact your Recruiting Partner.
Were building a new core group of engineers in TLV to drive innovation in compiler technology and systems-level ML software. As a Machine Learning Compiler Engineer youll help shape the direction of the team from the ground up and work on:
Designing and optimizing compiler features for high-performance ML workloads
Solving complex challenges in areas like graph optimization code generation and hardware-software co-design
Collaborating across Runtime Frameworks and Hardware teams to maximize end-to-end performance
As part of the Backend Compiler Team youll contribute to projects involving instruction scheduling memory management parallelism graph partitioning ISA design and new hardware bring-up.
This is a unique opportunity to be at the intersection of ML and systems within AWS helping to build the future of AI infrastructure right here in Tel Aviv.
Key job responsibilities
Our engineers collaborate across diverse teams projects and environments to have a firsthand impact on our global customer base. You will:
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
As you design and code solutions to help our team drive efficiencies in software architecture youll create metrics implement automation and other improvements and resolve the root cause of software defects. Youll also:
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 youre always working on the most important stuff.
Considering the technical requirements of this role and to ensure fair and equal candidate consideration we are conducting full interview loops. If you meet the basic qualifications and the team is interested in moving forward we will be in touch to coordinate your interviews.
About the team
*Diverse Experiences
Amazon values diverse experiences. Even if you do not meet all of the preferred qualifications and skills listed in the job description we encourage candidates to apply. If your career is just starting hasnt followed a traditional path or includes alternative experiences dont let it stop you from applying.
*Why AWS
Amazon Web Services (AWS) is the worlds most comprehensive and broadly adopted cloud platform. We pioneered cloud computing and never stopped innovating thats why customers from the most successful startups to Global 500 companies trust our robust suite of products and services to power their businesses.
*Work/Life Balance
We value work-life harmony. Achieving success at work should never come at the expense of sacrifices at home which is why flexible work hours and arrangements are part of our culture. When we feel supported in the workplace and at home theres nothing we cant achieve in the cloud.
*Inclusive Team Culture
AWS values curiosity and connection. Our employee-led and company-sponsored affinity groups promote inclusion and empower our people to take pride in what makes us unique. Our inclusion events foster stronger more collaborative teams. Our continual innovation is fueled by the bold ideas fresh perspectives and passionate voices our teams bring to everything we do.
*Mentorship and Career Growth
Were continuously raising our performance bar as we strive to become Earths Best Employer. Thats why youll find endless knowledge-sharing mentorship and other career-advancing resources here to help you develop into a better-rounded professional.
- 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
- 3 years of experience developing compiler optimization graph-theory hardware bring-up FPGA placement and routing algorithms or hardware resource management
- 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 is TensorFlow PyTorch and/or JAX
- Experience in LLM Vision or other deep-learning models
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 including support for the interview or onboarding process please visit for more information. If the country/region youre applying in isnt listed please contact your Recruiting Partner.
نوع التوظيف
دوام كامل
القسم / المجال المهني
المهارات المطلوبة
الإبلاغ عن هذه الوظيفة
إخلاء المسؤولية: د.جوب هو مجرد منصة تربط بين الباحثين عن عمل وأصحاب العمل. ننصح المتقدمين بإجراء بحث مستقل خاص بهم في أوراق اعتماد صاحب العمل المحتمل.
نحن نحرص على ألا يتم طلب أي مدفوعات مالية من قبل عملائنا، وبالتالي فإننا ننصح بعدم مشاركة أي معلومات شخصية أو متعلقة بالحسابات المصرفية مع أي طرف ثالث. إذا كنت تشك في وقوع أي احتيال أو سوء تصرف، فيرجى التواصل معنا من خلال تعبئة النموذج الموجود على الصفحة اتصل بنا
