Advanced Micro Devices, Inc

Sr. Staff Engineer, AI Models and Applications

Advanced Micro Devices, Inc • $150K — $200K *
San Jose, CA 95123In-Person
Consumer Technology
5 - 7 years of experience
Reposted Today
Be an Early Applicant
Job Overview by Ladders

Qualifications

  • PhD or master's degree in Computer Science, Electrical Engineering, Mathematics, or related field.
  • Strong technical knowledge of Generative AI applications and large language/multimodal models.
  • Experience in training models at scale.
  • Proficient with deep learning frameworks like PyTorch or JAX.
  • Demonstrated expertise in low-precision training and algorithmic innovations for AI.

Responsibilities

  • Propose innovative techniques for training and inferencing Generative AI models.
  • Implement efficient architectures for AI training and demonstrate performance on AMD devices.
  • Integrate AMD-optimized models into open-source frameworks and contribute to the community.
  • Collaborate with hardware and software teams for end-to-end performance optimization.
  • Promote and publish findings internally at AMD and in external venues.

Benefits

  • Opportunities to work at the forefront of AI innovation and technology.
  • Collaboration with a talented team of industry specialists.
  • Access to cutting-edge GenAI algorithms and systems.
  • Ability to influence the efficiency and scale of AI model training and inference.
Full Job Description
THE ROLE:

The AI Models and Applications team at AMD is looking for a specialized Sr. Staff or Principal level engineer who is passionate about enabling innovative and efficient Generative AI training/inferencing at scale. You will be part of a core team of incredibly talented specialists and work on scaling training and inference for the latest Generative AI models.

THE PERSON:

The ideal candidate has deep technical understanding of the latest generative AI applications like large language models (LLMs), large multimodal models (LMMs), image/video generation, has experience training models at scale and is passionate about innovating efficient approaches to enable distributed training and inference at scale on AMD devices.

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:

San Jose, CA (Hybrid).

Can also consider Seattle or Austin locations (Hybrid)

#LI-MV1

#LI-HYBRID

About Advanced Micro Devices, Inc

Advanced Micro Devices, Inc. Careers

Join the innovative forefront of technology with a career at Advanced Micro Devices, Inc. (AMD), a leader in semiconductor development. As part of our global team, you will contribute to an organization renowned for its dedication to innovation, leadership, and diversity in the tech industry.

Work You’ll Do

At AMD, we offer job opportunities that push the boundaries of what is possible. Our team is composed of professionals who lead the way in microprocessor and graphics technology, driving industry standards and innovation. With AMD, you will be part of a culture that values growth and professional development, ensuring that every team member has the opportunity to excel.

Transform Your Career

AMD is not just about advancing technology, but also about advancing careers. Whether you are looking for an internship, a full-time position, or leadership roles, AMD provides the platform to propel your career to new heights. Our commitment to professional growth is matched by our dedication to diversity and inclusion, making AMD a place where everyone can thrive.

Innovative Work Environment

Join a team of over 12,000 dedicated professionals at the intersection of technology, industry expertise, and digital innovation. At AMD, you will work on groundbreaking projects that shape the future of computing and graphics. Our collaborative environment encourages networking and the sharing of ideas across teams and disciplines.

Career Development and Benefits

AMD is committed to the development of its employees. We offer robust training programs, including leadership development and diversity training, to ensure our team is equipped for both current challenges and future opportunities. Our benefits package is designed to support the well-being and financial security of our employees and their families.

Explore Job Opportunities

From engineering to marketing, AMD offers a range of career paths that cater to diverse skills and interests. Our hiring process is designed to be transparent and engaging, helping you to understand where you fit within our team and how you can contribute to our collective goals.

Stay Connected

Join Our Team Search open positions that match your skills and interest. We look for passionate, curious, creative, and solution-driven team players. Explore the opportunities to join a company that’s committed to your career growth and to innovation in the technology sector.

Keep Up to Date

Stay ahead with career tips, insider perspectives, and industry-leading insights you can put to use today—all from the people who work here.

Job Alert Emails

Personalize your subscription to receive job alerts, latest news, and insider tips tailored to your preferences. Discover the exciting and rewarding career opportunities that await at Advanced Micro Devices, Inc.

Interview and Resume Tips

Prepare for your future with AMD by accessing resources that help you craft your resume and excel in interviews. Our goal is to help you showcase your best professional self and align your skills with the needs of our dynamic team. At Advanced Micro Devices, Inc., we empower our employees to innovate, lead, and grow. Join us in driving the future of technology while building a rewarding and sustainable career.
Learn more about Advanced Micro Devices, Inc
Size
15,500 employees
Market Cap
$100.9 billion
Industry
Manufacturing & Automotive
Net Income
$2.4 billion
Founded
1969
5 Year Trend
+30.9%
Revenue
$9.7 billion
NASDAQ
AMD
* Ladders Estimates

Similar Jobs

Ad banner background

Get Ready For Your
Next Interview

More Jobs at Advanced Micro Devices, Inc

More Consumer Technology Jobs

Find similar Sr. Staff Engineer, AI Models and Applications jobs:

NationwideSan Jose, CA