Senior Staff AI Framework Backend Engineer

Advanced Micro Devices View all jobs

  • Calgary, AB
  • Permanent
  • Full-time
  • 1 month ago
Job Description:WHAT YOU DO AT AMD CHANGES EVERYTHINGAt 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 ENGINEERTHE 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.KEY 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.
PREFERRED EXPERIENCE:
  • 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.
ACADEMIC CREDENTIALS:
  • Bachelor’s or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent
#LI-DR1Benefits offered are described: .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.AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD’s “Responsible AI Policy” is availableThis posting is for an existing vacancy.

Advanced Micro Devices