Tk Compiler Software Development Director

Jun 28, 2024
Santa Clara, Cuba
... Not specified
... Senior
Full time
... Office work


WHAT YOU DO AT AMD CHANGES EVERYTHING

We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world’s most important challenges. We strive for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives. 

AMD together we advance_




The Role

As a Tk compiler engineer, you will help design and implement the compiler infrastructure for a kernel language that can generate high performance kernels for a wide variety of ML use cases. Tk (Turbine Kernels) is a language that layers on top of the IREE compiler and runtime that can be used for writing high performance kernels for a wide variety of hardware backends.

 

The Person

A strong compiler and programming language expert, who is very familiar with the GPU/CPU programming model and understands the core optimization primitives. You are a strong communicator and team player.

 

Key Responsibilities

  • You will drive the design and development of the Tk language and compiler infrastructure
  • Implement compiler optimizations for generating high performance kernels on AMD GPUs, CPUs and IPUs
  • Implement state of the art ML kernels in Tk (such as paged attention, etc.) and optimize them
  • Participate in design discussions and develop roadmap for future work
  • Engage with the community and drive adoption of Tk for various use-cases

 

Preferred Experience

  • 5+ years of experience in compiler development
  • Strong understanding of programming language and compiler development
  • Experience with IREE, MLIR and/or LLVM
  • Experience with HPC programming and writing high performance kernels in CUDA, OpenCL, SYCL, etc.
  • Strong understanding of programming languages such as Triton, Mojo, Python, Haskell
  • Understanding of optimizations in generative models such as LLMs, SDXL etc.
  • Understanding of distributed computations and communication collectives

 

Academic Credentials

  • Masters degree or PhD in Computer Science or related field

#LI-EM1




At AMD, your base pay is one part of your total rewards package.  Your base pay will depend on where your skills, qualifications, experience, and location fit into the hiring range for the position. You may be eligible for incentives based upon your role such as either an annual bonus or sales incentive. Many AMD employees have the opportunity to own shares of AMD stock, as well as a discount when purchasing AMD stock if voluntarily participating in AMD’s Employee Stock Purchase Plan. You’ll also be eligible for competitive benefits described in more detail here.

 

AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law.   We encourage applications from all qualified candidates and will accommodate applicants’ needs under the respective laws throughout all stages of the recruitment and selection process.

At AMD, your base pay is one part of your total rewards package.  Your base pay will depend on where your skills, qualifications, experience, and location fit into the hiring range for the position. You may be eligible for incentives based upon your role such as either an annual bonus or sales incentive. Many AMD employees have the opportunity to own shares of AMD stock, as well as a discount when purchasing AMD stock if voluntarily participating in AMD’s Employee Stock Purchase Plan. You’ll also be eligible for competitive benefits described in more detail here.

 

AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law.   We encourage applications from all qualified candidates and will accommodate applicants’ needs under the respective laws throughout all stages of the recruitment and selection process.

The Role

As a Tk compiler engineer, you will help design and implement the compiler infrastructure for a kernel language that can generate high performance kernels for a wide variety of ML use cases. Tk (Turbine Kernels) is a language that layers on top of the IREE compiler and runtime that can be used for writing high performance kernels for a wide variety of hardware backends.

 

The Person

A strong compiler and programming language expert, who is very familiar with the GPU/CPU programming model and understands the core optimization primitives. You are a strong communicator and team player.

 

Key Responsibilities

  • You will drive the design and development of the Tk language and compiler infrastructure
  • Implement compiler optimizations for generating high performance kernels on AMD GPUs, CPUs and IPUs
  • Implement state of the art ML kernels in Tk (such as paged attention, etc.) and optimize them
  • Participate in design discussions and develop roadmap for future work
  • Engage with the community and drive adoption of Tk for various use-cases

 

Preferred Experience

  • 5+ years of experience in compiler development
  • Strong understanding of programming language and compiler development
  • Experience with IREE, MLIR and/or LLVM
  • Experience with HPC programming and writing high performance kernels in CUDA, OpenCL, SYCL, etc.
  • Strong understanding of programming languages such as Triton, Mojo, Python, Haskell
  • Understanding of optimizations in generative models such as LLMs, SDXL etc.
  • Understanding of distributed computations and communication collectives

 

Academic Credentials

  • Masters degree or PhD in Computer Science or related field

#LI-EM1

COMPANY JOBS
1155 available jobs
WEBSITE