Lead Machine Learning Engineer

AMD
Hyderabad
Workplace: OnsiteFull timeFunction: Data Science & Machine LearningEducation: high_schoolSkills: ["Collaboration","Cross-functional communication","Customer partnership","Problem-solving","Performance focus"]

Help deliver end-to-end AI/ML solutions for AMD’s next-generation compute platforms in Hyderabad. Work with advanced neural network architectures (CNNs, ViTs, LLMs, BERT, Transformers) and optimize them using AI/ML compiler toolchains and system-level engineering. Develop software frameworks and applications for ML workloads, and run performance analysis, simulations, profiling, and optimization in a collaborative R&D environment with hardware, software, application, and customer partners.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
AMD
AMD
1 month ago

Lead Machine Learning Engineer

✓ Verified Job

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

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

Job Summary

Help deliver end-to-end AI/ML solutions for AMD’s next-generation compute platforms in Hyderabad. Work with advanced neural network architectures (CNNs, ViTs, LLMs, BERT, Transformers) and optimize them using AI/ML compiler toolchains and system-level engineering. Develop software frameworks and applications for ML workloads, and run performance analysis, simulations, profiling, and optimization in a collaborative R&D environment with hardware, software, application, and customer partners.
Location: Hyderabad
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning

Key Responsibilities

  • •Work with AI/ML compiler toolchains across front-end, back-end, optimization, performance, and debugging domains.
  • •Collaborate with the AI/ML Solutions team to accelerate solution development and deployment.
  • •Evaluate and optimize neural network architectures (CNNs, ViTs, LLMs) for AMD platforms.
  • •Develop software frameworks, applications, and system-level solutions for ML workloads.
  • •Perform performance analysis, simulations, profiling, and optimization of AI applications.

Key Requirements

  • •Bachelor's or Master's degree in Computer Science, Computer Engineering, Data Science, Artificial Intelligence, or a related field.
  • •Experience with machine learning algorithms and deep learning frameworks.
  • •Strong understanding of deep neural network architectures such as CNNs, Vision Transformers (ViTs), and Transformer-based models.
  • •Hands-on experience with AI frameworks including ONNX, PyTorch, TensorFlow, and Caffe.
  • •Excellent C/C++/Python programming and software design skills, including debugging, performance analysis, and test design.
Education:High School
Skills:CollaborationCross-functional communicationCustomer partnershipProblem-solvingPerformance focus
Languages:English
Tech Stack:Machine LearningDeep LearningVision(CNN)Vision Transformers (ViTs)LLMsBERTTransformersONNXPyTorchTensorFlowCaffeResNetYOLOVision Transformer (ViT)BEVFormerCC++PythonCUDAHIP

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