Applied Research Engineer

Advanced Micro Devices, Inc.; California, San Jose (MO, CA); Hybrid; USD 120,000 - 180,000; Full time

Job summary

Join a forward-thinking company as an Applied Research Engineer, where your work will directly influence the future of AI technology. In this dynamic role, you'll explore and implement cutting-edge multimodal models, enhancing visual and audio understanding while accelerating training speeds. Collaborate with a world-class research team to push the boundaries of innovation and contribute to groundbreaking projects. This is an exciting opportunity to engage with academia and open-source communities, publish your findings, and shape the direction of AI platforms. If you are passionate about transforming lives through technology, this role is perfect for you.

Benefits

Health insurance

Retirement plan

Flexible working hours

Professional development opportunities

Remote work options

Wellness programs

Qualifications

• Expertise in training and fine-tuning large multimodal models.
• Knowledge of the latest ML techniques and model architectures.

Responsibilities

• Pre-train and post-train large multimodal models.
• Improve visual/audio understanding and modality fusion.

Skills

Training LMMs

Hyper-parameter tuning

Data preprocessing

Tokenization methods

Transformer architectures

Communication skills

Problem-solving skills

Education

PhD in Machine Learning

Master's in Computer Science

Tools

Python

PyTorch

JAX

TensorFlow

Job description

WHAT YOU DO AT AMD CHANGES EVERYTHING

We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world's most important challenges. We strive for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives.

AMD together we advance_

THE ROLE:

We are looking for Applied Research Engineer experienced with large multimodal foundation models. In this role, you will explore novel foundation models for visual/audio understanding with advanced LLMs 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, and aligning large multimodal models, in addition to keeping up-to-date to the latest progress and trends in large multi-modal model, unified visual foundation model and other foundation models.

THE PERSON:

Do you like to design and implement novel research ideas, improve the quality of the large multimodal models, accelerate the training and inference speed, 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 LMMs, familiar with hyper-parameter tuning techniques, data preprocessing, tokenization methods and latest training approaches.A successful candidate needs to be knowledgeable with the latest transformer architectures.

KEY RESPONSIBILITIES:

• Pre-train and post-train large multi-modality models.

• Improve visual/audio understanding, modality fusion and reasoning of multi-modality models

• Improve the state-of-the-art large multi-modality models.

• Accelerate the training and inference speed of LMMs

• 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 LMM pretraining, finetuning, and/or RLHF.

• 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:

Hybrid in San Jose, CA or Seattle WA preferred

#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.