Email Alert for
ⓘ There was an unexpected error processing your request.
Please refresh the page and try again.
If the problem persists, please contact us with your issue.
You can always manage your preferences and update your interests to ensure you receive the most relevant opportunities.
Get ready to discover your next great opportunity.
Meta
Menlo Park
**Summary:** In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new DL/ML model architectur
Job Source: MetaMeta
Menlo Park
**Summary:** In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new DL/ML model architectur
Job Source: MetaMeta
Menlo Park
**Summary:** In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new DL/ML model architectur
Job Source: MetaJuniper Networks
Sunnyvale, CA, United States
Job Description Staff Software Compiler Engineer (Compiler Team) Location: Sunnyvale, CA Juniper Networks seeks a highly skilled Software Staff Engineer to join our Software Engineering Compiler group. In this technical role, you will design, develop, and implement test plans and strategies to qualify and upgrade compiler toolchains for our cutt
Job Source: Juniper NetworksMeta
Menlo Park
**Summary:** The PyTorch Compiler team is dedicated to making PyTorch run faster and more resource-efficient without sacrificing its flexibility and ease of use. The team is the driving force behind PT2, a step function change in PyTorch’s history that brought compiler technologies to the core of PyTorch. PT2 technologies have gained industry-wide
Job Source: MetaConductor
San Jose, CA, United States
What You’ll Do The AGI (Artificial General Intelligence) Computing Lab is dedicated to solving the complex system-level challenges posed by the growing demands of future AI/ML workloads. Our team is committed to designing and developing scalable platforms that can effectively handle the computational and memory requirements of these workloads whil
Job Source: ConductorJuniper Networks
Sunnyvale, CA, United States
Job Description Staff Software Compiler Engineer (Compiler Team) Location: Sunnyvale, CA Juniper Networks seeks a highly skilled Software Staff Engineer to join our Software Engineering Compiler group. In this technical role, you will design, develop, and implement test plans and strategies to qualify and upgrade compiler toolchains for our cutt
Job Source: Juniper NetworksAMD
San Jose, CA, United States
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. Underpinn
Job Source: AMDSan Jose, CA, United States
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
AMD is looking for a world class AI frameworks and compiler engineer who can provide technical leadership in the development of various AI frameworks in the AMD ecosystem. You will need to drive technical direction for next generation frameworks for AI model training and inference for wide variety of AMD devices, current and future, such as MI Instinct, and Radeon GPUs, XDNA devices, including the recently released Ryzen AI, Alveo V70 and Versal ACAP, and datacenter CPUs such as EPYC. You will work enhance the AI framework capabilities to enable cutting-edge models on onto AMD’s cutting-edge hardware.
The Person
If you are passionate about AI/ML frameworks, software architecture, and/or compilers this is your opportunity. You will be working in one of the core areas, such as AI/ML frameworks components (e.g. PyTorch, TensorFlow, ONNX/OnnxRuntime), AI compiler and runtime, and/or optimization tooling to accelerate AI/ML workloads on AMD hardware platforms. You will collaborate closely with AI researchers to drive the development of framework components to efficiently map AI models to run on a variety of HW AI accelerators. You will have demonstrated focus in at least one of these areas: developing and deploying model optimization features, such as graph fusion, quantization, sparsity, and/or experience with compilers, possibly with compiler frameworks (e.g. LLVM, MLIR), and AI execution runtimes. You should be someone who can work in a dynamic, fast-paced development environment, with excellent leadership and collaboration skills. You will work with multiple engineering teams that are geographically dispersed. You will work on next generation framework software, guiding other senior developers and domain experts.
Required Experience
Experience with development in one of the focus areas: AI frameworks, AI compiler stack, and/or performance tuning and optimizations for workloads running on ML accelerator HWs.
Experience with ML frameworks such as PyTorch, OnnxRuntime, JAX, TensorFlow
Proficient in C++ programming.
Experience developing and debugging in Python.
Excellent skills at designing Python tools and libraries used by large number of users.
Experience with AI model architectures, e.g. Transformers, CNNs.
Team player and ready to work with a geographically distributed team.
Preferred Experience
Highly preferred - knowledge of custom accelerator hardware
Depth of experience with AI compilers, dataflow graphs, MLIR, and dialects, compiler optimizations, hardware specific optimizations.
Experience with MLIR, LLVM, IREE, XLA, TVM, Halide is a plus.
OR Depth of experience with model optimization, including quantization, sparsity.
Solid understanding of transformer architecture, LLMs, MoE.
Experience on LLMs/LMMs finetuning, distillation, and/or RLHF
ACADEMIC CREDENTIALS
BS, MS or PhD in Computer Science, Computer Engineering, Electrical Engineering, or related technical fields.
Location :
San Jose, Ca
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.
#J-18808-Ljbffr
Email Alert for Senior Staff Software Engineer - AI Frameworks / Compiler / Systems jobs in San Jose, CA, United States
ⓘ There was an unexpected error processing your request.
Please refresh the page and try again.
If the problem persists, please contact us with your issue.
You can always manage your preferences and update your interests to ensure you receive the most relevant opportunities.
Get ready to discover your next great opportunity.