Machine Learning Research Engineer Job in Bangalore at AMD

AMD is hiring for the role of Machine Learning Research Engineer!

Responsibilities of the Candidate:

• Pre-train and finetune over large GPU clusters while optimizing for various trade-offs. 
• Improve upon the state-of-the-art in Generative AI model architectures and training techniques. 
• Accelerate the training and inference speed across AMD accelerators. 
• Publish your work at top-tier conferences & workshops and/or through technical blogs. 
• Engage with academia and open-source ML communities. 
• Drive continuous improvement of infrastructure and development ecosystem.

Requirements:

• Strong development and debugging skills in Python. 
• Experience in deep learning frameworks (like PyTorch or TensorFlow) and distributed training tools (like DeepSpeed or Pytorch Distributed). 
• Experience with fine-tuning methods (like RLHF & DPO) as well as parameter-efficient techniques (like LoRA & DoRA).
• Solid understanding of various types of transformers and state space models. 
• Strong publication record in top-tier conferences, workshops or journals. 
• Solid communication and problem-solving skills