Responsibilities
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
Qualifications
Benefits offered are described: AMD benefits at a glance.
