Principal Engineer, Physical AI

AMD
Markham
Workplace: HybridFull timeFunction: Education & TrainingEducation: phdSkills: ["Communication","Problem-solving","Collaboration"]

Senior Principal-level engineer focused on scalable Generative AI training and inference. You will design and apply innovative training architectures and parallelism strategies, work with PyTorch/JAX and open-source frameworks, collaborate with software and hardware teams to co-optimize AMD solutions, and publish findings to advance Gen AI at scale across AMD devices.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
AMD
AMD
7 months ago

Principal Engineer, Physical AI

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 12 hours agoStatus: Live

Job Summary

Senior Principal-level engineer focused on scalable Generative AI training and inference. You will design and apply innovative training architectures and parallelism strategies, work with PyTorch/JAX and open-source frameworks, collaborate with software and hardware teams to co-optimize AMD solutions, and publish findings to advance Gen AI at scale across AMD devices.
Location: Markham
Workplace: Hybrid
Employment Type: Full time
Job Function: Education & Training
Seniority: Sr. Manager level

Key Responsibilities

  • •Propose and apply innovative techniques to support both training and inferencing including transformer architectures, parallelism strategies to train on large clusters, and low-precision training.
  • •Implement novel efficient architectures for Generative AI models for training and inference and showcase benefits on AMD.
  • •Work with open-source framework and community (e.g., PyTorch, JAX, Hugging Face) to integrate AMD optimized models, libraries and publish training recipes.
  • •Collaborate with software and hardware teams to E2E co-optimize performance on current and future AMD solutions.
  • •Publish and promote your work within AMD and at external venues.

Key Requirements

  • •PhD or master's degree (or higher) in CS, EE, Mathematics, or a related field.
  • •Extensive experience with Generative AI model training and inference and familiarity with PyTorch or JAX.
  • •Experience training models at scale and distributed training/inference on AMD hardware.
  • •Strong publication record in relevant conferences (e.g., NeurIPS, CVPR, ICML, ICLR) is a plus.
  • •Excellent written, verbal, and presentation skills; ability to coordinate internally and externally.
Education:PhD / Doctorate
Skills:CommunicationProblem-solvingCollaboration
Languages:English
Tech Stack:PyTorchJAXHugging Face

Company Brief

AMD
Designs and produces semiconductor products including CPUs, GPUs, and adaptive SoCs for consumer, enterprise, and embedded markets, competing across PCs, data centers, and gaming industries.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1969
Glassdoor
Glassdoor: 3.9
WebsiteLinkedIn