Applied Research Scientist - AI Models & Agents

Overview

Improve AI model architectures and performance.

Responsibilities

• Improve upon the state-of-the-art in Generative AI model architectures and their compatibility on AMD accelerators
• Accelerate the training and inference speed through various optimizations.
• Build AI agents to automatically write and evaluate efficient kernels and code
• Drive continuous improvement of infrastructure and development ecosystem
• Publish your research at top-tier conferences, workshops and/or through technical blogs.
• Engage with academia and open-source ML communities.
• Strong development and debugging skills in Python.

Qualifications

• Solid understanding of various types of transformers and state space models.
• prior experience in building self-evolving AI agents for various challenging tasks (like code generation, discovery, etc.)
• knowledge of latest research in the field of AI agents
• familiar with techniques to do hardware-efficient training and inference of large language models and also familiarity with existing frameworks (e.g. vllm, sglang, etc.)
• expertise in model architecture (beyond just transformers only models) and latest research in this field
• having relevant open-source contributions
• Strong publication record in top-tier conferences, workshops or journals.
• Solid communication and problem-solving skills.
• Passionate about learning new stuffs in this domain as well as innovating on top of it
• Advanced degree (Master’s or PhD) in machine learning, computer science, artificial intelligence, or a related field is expected. Exceptional Bachelor’s degree candidates with years of relevant research experience may also be considered.

Benefits

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