GPU ML Inference Engineer

AMD
  • Warsaw
  • Post Date: October 6, 2024
    • 14549
  • Applications 0
  • Views 3
Job Overview

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_

GPU ML Inference Engineer

The Role

AMD is looking for a strategic ML Engineer who is passionate about writing fast GPU kernels for Machine Learning inference in the context of image processing and real-time 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.

The Person

The ideal candidate is a highly skilled software or machine learning engineer, with expertise in low level GPU programming. You are interested in Machine Learning, and you are excited about the opportunities that real time inference 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

Writing Machine Learning inference code which is fast and efficient on GPUs.Ability to work with D3D12 and HLSL.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 solutionsDocument 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) experience efficiently creating C/C++ game code for Windows and knows all about data structures, design patterns, language features, standard libraries and writing easy-to-understand codeFamiliarity with 3D graphics and ray tracing techniques using GPU Compute and popular graphics APIs such as Direct3D, Vulkan, OpenGL, OpenCL, CUDA, and HIP.Has strong graphics code optimization skills, in particular shader code optimizations.Ability to write high-quality, maintainable code with meticulous attention to detail, ensuring robustness and performance optimizationExperience with modern concurrent programming paradigms and threading APIs to develop parallel and distributed machine learning algorithms efficientlyEffective 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 experienceExperience with ML tools such as ONNX, PyTorch and similarGame engine or game rendering framework experience

Academic Credentials

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

LOCATION: Sweden, Germany, Poland, UK, Holland

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.

Job Detail
Shortlist Never pay anyone for job application test or interview.

Related Jobs (3230)

Safety Information

  • Safety Tips For Candidate

    DYOR.

  • Safety Tips For Candidate

    Always check the employer\'s offer.