Principal Engineer - Gen AI training and inferencing

KEY RESPONSIBILITIES:

Propose and apply innovative techniques to support both training and inferencing including innovative transformer architectures, parallelism strategies to train on large clusters, 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 team to E2E co-optimize performance on current and future AMD solutions.

Publish and promote your work within AMD and at external venues.

PREFERRED EXPERIENCE:

Strong technical expertise in Gen AI model training and inference, and familiarity working with deep learning frameworks like PyTorch/JAX.

Strong technical expertise in algorithmic innovation towards efficient Gen AI application for both training and inferencing.

Expertise/publications in one of the areas preferred - efficient model architectures, optimized training, innovative parallelism strategies or low-precision training.

Additional plus if publications include conferences such as NeuRIPS, CVPR, ECCV/ICCV, ICML, ICLR, etc.

Experience productizing generative AI models and training foundation model at Scale

Excellent written, verbal, and presentation skills, ability to coordinate internally and externally.

Several years of experience in AI, deep learning and related software development.

ACADEMIC CREDENTIALS:

PhD or master’s degree or above; with major in CS, EE, Mathematics, or a related field.

LOCATION:

Markham, Ontario, Canada (Hybrid) Can also consider Calgary, Alberta, Canada

#LI-MV1

#HYBRID

Benefits offered are described: AMD benefits at a glance .

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 available here.

This posting is for an existing vacancy.