Job Update: Computer Science Graduates, Postgraduates Vacancy at AMD

Overview:

AMD is hiring eligible and interested applicants for the post of Principal ML Engineer at their Bangalore, India location. 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.

To complete details for this job are as follows.

Roles & 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

• 10+ years of 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

Disclaimer: The Recruitment Information provided above is for informational purposes only. The above Recruitment Information has been taken from the official site of the Organisation. We do not provide any Recruitment guarantee. Recruitment is to be done as per the official recruitment process of the company or organization posted the recruitment Vacancy. We don’t charge any fee for providing this Job Information. Neither the Author nor Studycafe and its Affiliates accepts any liabilities for any loss or damage of any kind arising out of any information in this article nor for any actions taken in reliance thereon.