Senior Staff AI Framework Backend Engineer

Advanced Micro Devices, Inc•Calgary, AB

69d

About The Position

At AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you’ll discover the real differentiator is our culture. We push the limits of innovation to solve the world’s most important challenges—striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond. Together, we advance your career. SMTS - AI FRAMEWORKS BACKEND INTEGRATION ENGINEER THE ROLE: AMD is seeking an experienced AI Backend Integration Engineer to drive the performance and scalability of our deep learning platforms. This role involves working at the intersection of high-performance computing and machine learning frameworks, ensuring optimized execution of AI models across various hardware architectures. The ideal candidate will have deep expertise in systems programming (C++/Python), a solid understanding of parallel computing principles, hands-on experience working with deep learning libraries like MIOpen, hipDNN, or equivalent libraries such as cuDNN, and experience working in the backend integration of a framework like PyTorch, JAX, or ONNX Runtime. THE PERSON: The ideal candidate should be passionate about software engineering and possess leadership skills to drive sophisticated issues to resolution. Able to communicate effectively and work optimally with different teams across AMD. Comfortable navigating large, multi-team codebases for dealing with the ambiguity where the code boundaries meet.

Requirements

Expertise in C++ and Python is essential for performance-critical applications and general AI development.
Strong hands-on experience with major deep learning frameworks such as PyTorch and TensorFlow including their backend integrations.
Strong understanding of system performance, parallel computing principles, and memory management.
Experience with Linux and Windows operating system development
Familiarity with modern software development tools and practices, including version control (Git) and CI/CD pipelines.
Applied experience using AI‑assisted coding tools in professional software engineering workflows, including code generation, refactoring, test creation, documentation, and design exploration.

Responsibilities

Design, develop, and implement the integration of low-level backend libraries (e.g., cuDNN, MIOpen, hipDNN) into high-level AI frameworks like PyTorch and TensorFlow.
Identify and resolve complex issues related to model execution, data flow, and performance bottlenecks within the integrated systems.
Implement best software engineering practices, including code reviews, comprehensive testing (unit and integration), and detailed technical documentation for all integrations and optimizations.
Work closely with data scientists, ML engineers, and other software development teams to translate research ideas and model requirements into robust, production-ready systems.
Stay abreast of the latest advancements in AI/ML research, GPU technology, and related software ecosystems to propose and implement cutting-edge solutions.
Leverage AI‑assisted development tools to accelerate analysis, integration validation, and documentation, while ensuring correctness and traceability.