GPU ML Modeling/MLOps Engineer

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

AMD is looking for a strategic ML Engineer who is passionate about Machine Learning, Visual Computing, Gaming and Machine Learning.

The Person

The ideal candidate is a highly skilled software or machine learning engineer, with expertise in training models especially in computer vision aspect. You are interested in Machine Learning, and you are excited about the opportunities that have real impact on future gaming environment can unlock for the next generation of software technologies applied to graphics. You are team-driven and motivated to do things others might find too difficult.

Key Responsibilities

• Modeling, training and maintaining ML CV models.
• Proficient in Python for rapid prototyping and testing.
• Comprehensive understanding of general software development workflows, including version control systems like Git, build automation tools like CMake, and continuous integration (CI) pipelines.
• Proficient in English, with excellent written and verbal communication skills for collaborating with team members and presenting findings or proposals.
• Collaborate with cross-functional teams including researchers, engineers, and domain experts to understand requirements, develop prototypes, and deploy production-ready machine learning inference solutions
• Document code, methodologies, and findings comprehensively, ensuring reproducibility and knowledge sharing within the team and across the organization.
• Mentor junior team members, providing guidance on machine learning concepts, programming techniques, and software development practices to foster skill development and team growth.

Preferred Experience

• Minimum 2 years of full-time (post grad) on Data scientist/analyst position.
• Experience with ML tools such as ONNX, PyTorch and similar
• 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
• Effective communication and problem-solving skills, with the ability to articulate complex technical concepts to both technical and non-technical stakeholders.

The Following Experiences Are Optional But Highly Desired

• GCN / RDNA graphics or game console architecture experience
• Game engine or game rendering framework experience

Academic Credentials

• Bachelor’s or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent

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.

_