ML Kernel Software Development Engineer

Join Amd As A New College Graduate

What you do at AMD changes everything. 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. Together, we advance your career.

The Role

We are seeking a talented and motivated new college graduate to join our ROCm Core team. You will be responsible for enhancing and optimizing ML kernels to improve performance and efficiency. This is an exciting opportunity to work with cutting-edge technologies and contribute to innovative ML solutions.

The Person

The ideal candidate possesses an innovative and problem-solving mindset, has a keen eye for software engineering development, and is diligent and passionate about technology.

Key Responsibilities

Develop and optimize ML kernels with focus on performance and scalability. Collaborate with cross-functional teams to understand requirements and deliver high-quality software components. Assist in design and implementation of new ML algorithms and frameworks. Participate in code reviews, testing, and debugging processes to ensure the delivery of reliable software. Keep up-to-date with the latest advancements in machine learning and hardware optimization techniques.

Preferred Experience

Strong foundation in machine learning concepts and algorithms. Experience in programming languages such as C++, Python, or similar. Familiarity with GPU programming and parallel computing, like CUDA or ROCm. Experience with ML frameworks like TensorFlow, PyTorch, or similar. Excellent problem-solving skills and attention to detail. Ability to work in a team-oriented environment and excellent communication skills.

Academic Credentials

Bachelor or above degree in Computer Science, Electrical Engineering, or related field.

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.