Applied Research Scientist - AI Models & Agents

AMD
Bengaluru
Workplace: OnsiteFull timeINR 3,052,140 - 4,360,200 annuallyFunction: Data Science & Machine LearningEducation: mastersSkills: ["Communication","Problem-solving","Collaboration","Leadership"]

Join AMD's AI Models team to advance training and inference for large language models (LLMs), large multimodal models (LMMs), image/video generation, and self-evolving agents. You will contribute to scalable pre-training, instruction tuning, and alignment, shaping strategy as an early team member. Expect hands-on Python work with PyTorch/TensorFlow, kernel optimization (HIP/CUDA/Triton), and collaboration with world-class researchers and academia.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
AMD
AMD
9 months ago

Applied Research Scientist - AI Models & Agents

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 12 hours agoStatus: Live

Job Summary

Join AMD's AI Models team to advance training and inference for large language models (LLMs), large multimodal models (LMMs), image/video generation, and self-evolving agents. You will contribute to scalable pre-training, instruction tuning, and alignment, shaping strategy as an early team member. Expect hands-on Python work with PyTorch/TensorFlow, kernel optimization (HIP/CUDA/Triton), and collaboration with world-class researchers and academia.
Location: Bengaluru
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning

Key Responsibilities

  • •Improve state-of-the-art generative AI model architectures and their compatibility on AMD accelerators
  • •Accelerate training and inference speed through various optimizations
  • •Build AI agents to automatically write and evaluate efficient kernels and code
  • •Drive continuous improvement of infrastructure and development ecosystem
  • •Publish research at top-tier conferences, workshops, or through technical blogs and engage with academia/open-source ML communities

Pay and Benefits

Salary: INR 3,052,140 - 4,360,200 annually

Key Requirements

  • •Advanced degree (Master’s or PhD) in machine learning, computer science, artificial intelligence, or a related field
  • •Hands-on experience with developing and optimizing LLMs, LMMs, and/or diffusion models
  • •Strong development and debugging skills in Python
  • •Experience with deep learning frameworks (e.g., PyTorch or TensorFlow) and distributed training tools
  • •Experience writing kernels using HIP, CUDA, Triton, etc.
Experience:AI/MLResearch & Development
Education:Master's
Skills:CommunicationProblem-solvingCollaborationLeadership
Languages:English
Tech Stack:PythonPyTorchTensorFlowHIPCUDATritonDistributed training

Company Brief

AMD
Designs and produces semiconductor products including CPUs, GPUs, and adaptive SoCs for consumer, enterprise, and embedded markets, competing across PCs, data centers, and gaming industries.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1969
Glassdoor
Glassdoor: 3.9
WebsiteLinkedIn