Staff Graphics Research Engineer (AI/ML)

Join to apply for the Staff Graphics Research Engineer (AI/ML) role at AMD.

Overall Mission and Culture

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.

The Role

AMD is looking for a strategic ML Research Engineer who is passionate about creating new experiences with GPU rendering. You will be a member of a core team of incredibly talented industry specialists and will work with the very latest hardware and software technology to push the boundaries of what is possible in rendering.

Key Responsibilities

• Expertise in Machine Learning, particularly focused on Model Creation and Model Architecture, including advanced techniques such as deep learning, reinforcement learning, and generative models.
• Proficiency in 3D graphics and Ray tracing using GPU Compute and Graphics APIs like Direct3D, Vulkan, OpenGL, OpenCL, CUDA.
• Knowledge in denoising techniques.
• Expertise in either Diffusion or Generative Transformer based LLM models.
• Strong proficiency in Python programming for implementing machine learning algorithms, data preprocessing, and model evaluation.
• Comprehensive understanding of general software development workflows, including version control systems like Git, environment management tools like Docker, Conda, and continuous integration (CI) pipelines.
• Proficient in English, with excellent written and verbal communication skills.
• Collaborate with cross‑functional teams to understand requirements, develop prototypes, and deploy production‑ready machine learning solutions.
• Research and stay up‑to‑date with the latest advancements in machine learning, frameworks, and tools.
• Optimize machine learning models for deployment on various platforms including cloud infrastructure, edge devices, and embedded systems.
• Conduct thorough experiments and evaluations to assess model performance, reliability, and robustness, employing techniques such as hyperparameter tuning, cross‑validation, and A/B testing.
• Document code, methodologies, and findings comprehensively, ensuring reproducibility and knowledge sharing.

Preferred Experience

• Extensive knowledge and hands‑on experience in machine learning, with a track record of successfully creating and optimizing machine learning models for various applications, especially around ML Model Architectures.
• Demonstrated expertise in designing efficient and scalable model architectures tailored to specific problem domains or computational resources.
• Familiarity with 3D graphics and ray tracing techniques using GPU Compute and popular graphics APIs such as Direct3D, Vulkan, OpenGL, OpenCL, CUDA, and HIP.
• Ability to write high‑quality, maintainable code with meticulous attention to detail, ensuring robustness and performance optimization.
• Experience with modern concurrent programming paradigms and threading APIs to develop parallel and distributed machine learning algorithms efficiently.
• Proficiency in both Windows and Linux operating system development environments.
• Familiarity with software development processes and tools such as debuggers, source code control systems (e.g., GitHub), and performance profilers.
• Strong programming skills in C++ for implementing performance‑critical components of machine learning frameworks or applications.
• Effective communication and problem‑solving skills, with the ability to articulate complex technical concepts to both technical and non‑technical stakeholders.
• Demonstrated leadership qualities and interpersonal skills, capable of motivating and guiding team members to achieve project goals effectively.

Academic Credentials

Bachelor’s or Master’s degree in Computer Science, with a focus on areas such as Graphics, Mathematics, Machine Learning, Computer Engineering, or related fields, providing a solid theoretical foundation for advanced machine learning research and development.

Benefits

Benefits are described: AMD benefits at a glance.

Equal Opportunity, Inclusive Employment Statement

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.

Seniority Level

Mid‑Senior level

Employment Type

Full‑time

Job Function

Semiconductor Manufacturing