Search jobs > San Jose, CA > Ai ml engineer

Principal AI/ML Compiler Engineer

Advanced Micro Devices, Inc
San Jose, California, United States
Full-time

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 : In this team you will be building the compiler technology used to accelerate the latest AI models on AMD CPUs addressing the areas such as vision models, speech recognition, working with the leading engineers in AMD’s CPU, GPU and Adaptable Compute teams THE PERSON : You will be someone with great depth of experience in compilers and compiler frameworks such as TVM, MLIR, LLVM and be read to hit the ground running writing some of the key components of the software involved in mapping AI / ML models from frameworks such as PyTorch, TensorFlow, TVM, MLIR.

You are someone that has insight into performance optimizations and will be able to think of novel solutions, implementing them from concept to production.

KEY RESPONSIBILITIES : Develop the latest algorithms, software, and architectures for AI acceleration. In this role you will be a technology leader and provide expertise to help AMD build the best AI acceleration solutions in the industry.

The focus will be on delivering the compiler software needed for targeting AMD's latest AI acceleration capabilities in our CPUs.

PREFERRED EXPERIENCE : Knowledge of AI frameworks like TensorFlow, PyTorch. Proven experience delivering compilers in TVM or MLIR Understanding of AI application stacks especially in LLM, ASR, Generative AI.

Knowledge of Acceleration platforms like GPU, TPU, APU, FPGAs. Proven experience in Software Development delivering C++ applications ACADEMIC CREDENTIALS : M.

S. or Ph.D #LI-JT1 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 : In this team you will be building the compiler technology used to accelerate the latest AI models on AMD CPUs addressing the areas such as vision models, speech recognition, working with the leading engineers in AMD’s CPU, GPU and Adaptable Compute teams THE PERSON : You will be someone with great depth of experience in compilers and compiler frameworks such as TVM, MLIR, LLVM and be read to hit the ground running writing some of the key components of the software involved in mapping AI / ML models from frameworks such as PyTorch, TensorFlow, TVM, MLIR.

You are someone that has insight into performance optimizations and will be able to think of novel solutions, implementing them from concept to production.

KEY RESPONSIBILITIES : Develop the latest algorithms, software, and architectures for AI acceleration. In this role you will be a technology leader and provide expertise to help AMD build the best AI acceleration solutions in the industry.

The focus will be on delivering the compiler software needed for targeting AMD's latest AI acceleration capabilities in our CPUs.

PREFERRED EXPERIENCE : Knowledge of AI frameworks like TensorFlow, PyTorch. Proven experience delivering compilers in TVM or MLIR Understanding of AI application stacks especially in LLM, ASR, Generative AI.

Knowledge of Acceleration platforms like GPU, TPU, APU, FPGAs. Proven experience in Software Development delivering C++ applications ACADEMIC CREDENTIALS : M.

S. or Ph.D #LI-JT1At 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.

7 days ago
Related jobs
Promoted
Tykhe Inc (pronounced Tie-key)
San Jose, California

We are seeking a Principal AI Research Engineer for conducting research and development in Generative AI, with a focus on creating and improving algorithms and models focused on multiple modalities such as text, images, and time-series data. If you have a proven track record of research in any area ...

Promoted
Percipient.ai
Santa Clara, California

The Principal Back-End Software Engineer will work collaboratively with other back-end engineers, front-end engineers, product managers, and AI and Computer Vision scientists to build scalable and innovative back-end systems, services, tools, and APIs that power our product. We pride ourselves in ma...

Promoted
Adobe
San Jose, California

AI/ML Intern - Machine Learning Engineer. Search, Discovery, and Content AI. Adobe is looking for a Machine Learning intern who will apply AI and machine learning techniques to real business problems to help Adobe better understand, lead and optimize the experience of its customers. By utilizing lat...

Oracle
Santa Clara, California

Oracle Cloud Infrastructure (OCI) Cluster Networking team is building an ultra-high performance network required to support AI/ML/HPC workloads. This team will be responsible for designing, developing and performance tuning the networking stack required to run distributed AI/ML/HPC workload across t...

Palo Alto Networks
Santa Clara, California

The AI security cloud service engineering team is the core engineering team to build a solid product to assure the runtime security of our customers when they are using AI especially LLM services. Palo Alto Networks is committed to AI security in the emerging AI era. Collaborate with product manager...

NVIDIA
Santa Clara, California

Strong background in software engineering, with experience in building and maintaining large-scale distributed systems, preferably in the context of AI/ML infrastructure. Collaborate with diverse teams, including researchers, data engineers, and DevOps professionals, to create a seamless and integra...

Waymo
Mountain View, California

ML Compiler Engineer, Compute. Enhanced leave options include paid parental leave (birthing parent gets 24 weeks of paid leave with up to 4 weeks of additional leave before their due date, and non-birthing parent gets 18 weeks of paid leave), and 20 days of subsidized backup childcare or adult/elder...

NVIDIA
Santa Clara, California

Understanding of the functioning of AI services, deep learning and AI. AI Applications organization to work on distributed cloud native accelerated video analytics applications. Our team is building distributed cloud native accelerated real-time video streaming AI inference and video analytics platf...

Annapurna Labs (U.S.) Inc.
Cupertino, California

The ML Distributed Training team works side by side with chip architects, compiler engineers and runtime engineers to create , build and tune distributed training solutions with Trn1. The Annapurna Labs team at Amazon Web Services (AWS) is looking for a Software Development Engineer II to build, del...

Clear Ventures
California, US

Ready to join a team that’s helping create a more sustainable, intelligent, and efficient world? Learn more at PLEASE READ THE REQUIREMENTS FOR THIS ROLE – ONLY CANDIDATES WITH SUPPLY CHAIN OPERATIONS/STRATEGY CONSULTING EXPERIENCE AND TECHNOLOGY / PRODUCT DELIVERY EXPERIENCE CAN BE CONSIDERED – THI...