Applied Research Scientist

CareerArc; Markham; On-site; CAD 120,000 - 180,000; Full time

Job summary

AMD in Markham, Ontario (Hybrid) is seeking an Applied Research Scientist focused on training large language and multimodal models, including diffusion-based Image/Video generation. You will explore clustering architectures, scale up pre-training and RL, and align models with performance and efficiency goals.

The ideal candidate has deep experience with LLMs/LMMs, transformers, Python, and major ML frameworks, and will publish contributions to advance AI platforms at AMD.

Qualifications

• Proven experience with training LLMs/LMMs and diffusion models.
• Experience with hyper-parameter tuning and data preprocessing.
• Familiar with tokenization methods and latest training approaches for foundation models.
• Strong knowledge of transformer architectures and model scaling.
• Proficient in Python and ML frameworks (PyTorch, JAX, TensorFlow).

Responsibilities

• Train and fine-tune LLMs, LMMs, and image/video generation models.
• Advance state-of-the-art through novel ML techniques and architectures.
• Accelerate training and inference speed for large models.
• Publish work at top-tier venues and engage with the ML community.
• Influence the direction of AMD AI platform and collaborate with academia.

Job description

WHAT YOU DO AT AMD CHANGES EVERYTHING

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.

THE ROLE:

We are looking for Applied Research Scientist experienced with training large language models, large multimodal models, and image/video generation models. In this role, you will explore novel LLM/LMM and image/video generation architectures and large-scale training techniques to advance the state-of-the-arts. You will be part of a world-class research team working on pre-training, fine-tuning, RL, and aligning large language models, large multimodal models, and image/video generation models, in addition to keeping up-to-date to the latest progress and trends in LLM/LMM, image/video generation, and other foundation models.

THE PERSON:

Do you like to design and implement novel research ideas, improve the quality of the large language/multimodal models, accelerate the training and inference speed of LLMs, LMMs, and image/video generation models, and influence future hardware and software direction? If so, this role is for you.The ideal candidate will have expertise and hands‑on experience on training LLMs, LMMs, and/or diffusion models., familiar with hyper-parameter tuning techniques, data preprocessing, tokenization methods and latest training approaches for LLMs, LMMs, and diffusion models.A successful candidate needs to be knowledgeable with latest transformer architectures.

KEY RESPONSIBILITIES:

• Train and finetune LLMs, LMMs, and image/video generation models.
• Improve on the state-of-the-art LLMs, LMMs, and image/video generation models.
• Accelerate the training and inference speed of LLMs, LMMs, and image/video generation models.
• Research novel ML techniques and model architectures.
• Influence the direction of AMD AI platform.
• Publish your work at top-tier venues.
• Engage with academia and open-source ML communities.

PREFERRED EXPERIENCE:

• Experience in developing and debugging in Python.
• Experience in ML frameworks such as PyTorch, JAX or TensorFlow.
• Experience with distributed training.
• Expertise on LLM/LMM/Diffusion pretraining, finetuning, and RL.
• Familiar with transformer architecture.
• Strong communication and problem-solving skills.
• Publication at top-tier venues is a huge plus.

ACADEMIC CREDENTIALS:

A PhD or master's degree or equivalent in machine learning, computer science, artificial intelligence, or a related field.

LOCATION:

Markham, Ontario (Hybrid)

#LI-MV1

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.