Software Development Engineer

7 hours ago


Chaoyang District, China Advanced Micro Devices, Inc Full time


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 looking for Machine Learning Engineer to join our Models and Applications team. If the challenge of distributed training of large model on large number of GPUs excites you and you are passionate about improving training efficiency and enjoy innovating and coming up with new ideas, then this role is for you.

You will be part of world class team focus on addressing the challenge of training generative AI.

THE PERSON:

The ideal candidate should have experience with distributed training pipeline, knowledgeable with distributed training algorithms (Data parallel, Tensor parallel, Pipeline parallel, ZeRO) and familiar with training Large Model.

KEY RESPONSIBILITIES:

  • Train large model to convergence on AMD GPUs.
  • Improve the end-to-end training pipeline performance.
  • Optimize the distributed training pipeline and algorithm to scale out.
  • Contribute your changes to open source.
  • Up to date with latest training algorithms.
  • Influence the direction of AMD AI platform.
  • Cross team collaborate with various group and stakeholder.

PREFERRED EXPERIENCE:

  • Experience in ML frameworks such as PyTorch, JAX or Tensorflow.
  • Experience with distributed training and distributed training framework such as DeepSpeed.
  • Experience with LLM or Vision, especially large model is a plus.
  • Excellent python programing skills, including debugging, profiling, and perf analysis.
  • Experience with ML pipeline.
  • Strong communication and problem-solving skills.

ACADEMIC CREDENTIALS:

A master's degree in computer science, artificial intelligence, machine learning, or a related field.

LOCATION:

Beijing or Shanghai

#LI-FL1



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.



  • Chaoyang District, China Advanced Micro Devices, Inc Full time CN¥120,000 - CN¥180,000 per year

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


  • Chaoyang District, China Blue Cross and Blue Shield of Louisiana Full time CN¥600,000 - CN¥1,200,000 per year

    We take great strides to ensure our employees have the resources to live well, be healthy, continue learning, develop skills, grow professionally and serve our local communities. We invite you to apply for a career with Blue Cross.Residency in or relocation to Louisiana is preferred for all positions. Only candidates located in Louisiana or willing to...


  • Panyu District, Guangdong, China Li & Fung Full time CN¥900,000 - CN¥1,200,000 per year

    Are you a movement maker? Are you seeking new and exciting career opportunities?Here is what you need to know about the job:We are looking for an experienced software development engineer to join our team in Futian, Shenzhen. You will be responsible for designing, developing, and maintaining enterprise-grade applications based on Oracle ADF, Java, and...


  • Chaoyang District, China Blue Cross and Blue Shield of Louisiana Full time $120,000 - $180,000 per year

    We take great strides to ensure our employees have the resources to live well, be healthy, continue learning, develop skills, grow professionally and serve our local communities. We invite you to apply for a career with Blue Cross.Residency in or relocation to Louisiana is preferred for all positions. POSITION PURPOSEIndividuals within the Senior Software...


  • Chaoyang District, China Blue Cross and Blue Shield of Louisiana Full time $80,000 - $120,000 per year

    We take great strides to ensure our employees have the resources to live well, be healthy, continue learning, develop skills, grow professionally and serve our local communities. We invite you to apply for a career with Blue Cross.Residency in or relocation to Louisiana is preferred for all positions. POSITION PURPOSEIndividuals within the Senior Software...


  • Chaoyang District, China Thales Full time CN¥100,000 - CN¥200,000 per year

    Location: Beijing, ChinaThales is a global technology leader trusted by governments, institutions, and enterprises to tackle their most demanding challenges. From quantum applications and artificial intelligence to cybersecurity and 6G innovation, our solutions empower critical decisions rooted in human intelligence. Operating at the forefront of aerospace...


  • Minhang District, China Detection Technology Full time $60,000 - $120,000 per year

    With us You doWORK THAT MATTERSWe are the most trusted partner imaging the unknown by creating foresight capabilities quicker to discover and prevent threats. Our solutions contribute to the improvement of people's health and safety daily. With us you do work that truly matters.We are a global provider of X-ray detector solutions and services for medical,...


  • Dongcheng District, , China Everbridge Full time CN¥120,000 - CN¥240,000 per year

    TITLE:  Senior Web Software DeveloperLOCATION:  Beijing, ChinaAbout the Team:As a member of our engineering team in Beijing Office, you will join a team of bright, fast paced, and outstanding developers. Within Engineering Team in Beijing office, you'll work on building cutting edge technology applications that will power our company's impressive growth....


  • Dongcheng District, , China Everbridge Full time $100,000 - $150,000 per year

    Everbridge, Inc. is seeking a talented  Web Software Developer. The ideal candidate must be proficient in the development of based high-performance Java applications. The position requires excellent problem-solving skills along with excellent communication skills.About the Team:As a member of our engineering team in Beijing Office, you will join a team of...


  • Chaoyang District, China WirelessCar Full time CN¥300,000 - CN¥600,000 per year

    Service Reliability EngineerWirelessCar's JourneyWirelessCar drives the future of mobility by connecting vehicles and developing cutting-edge digital services. We help leading brands—like Volkswagen, Volvo, and Jaguar Land Rover—innovate, enhance mobility, and accelerate their digital transformation. Join us and shape the next era of automotive...