AI/ML Solutions Engineer – LLM & Agent Development

AI/ML Solutions Engineer – LLM & Agent Development

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.

AMD is looking for a sharp, fast-moving AI Tooling Engineer to help us build intelligent assistants and automation tools for source code understanding, performance analysis, and developer productivity, all powered by cutting-edge LLM APIs.

The ideal candidate is a proactive problem-solver with strong technical expertise and a passion for AI-driven innovation. The successful candidate for this position will be interacting with software and hardware technologists working across many locations. This is a great opportunity to work as a part of highly regarded team to deliver leading AI-edged solutions.

Key Responsibilities:

• Build and maintain AI assistants using LangChain or LlamaIndex for source code, logs, and documentation.
• Develop intuitive UIs and dashboards using Streamlit, Gradio, or lightweight React-based apps.
• Implement Retrieval-Augmented Generation (RAG) pipelines with contextual grounding.
• Create VSCode extensions or integrations using Model Context Protocol (MCP).
• Deploy reliable services via FastAPI/Flask and host them on internal platforms.
• Work cross-functionally with ML, DevTools, and performance teams to iterate on prompt design, API efficiency, and tool ergonomics.
• Perform data analysis and visualization using Matplotlib, NumPy, and Scikit-Learn; parse logs (XML/JSON), apply Regex, and extract performance metrics.

Required Skills:

• Strong Python and JavaScript fundamentals
• Ability to work with REST APIs, manage tokens, and handle context-sensitive requests
• Experience with LangChain, LlamaIndex, or similar LLM orchestration frameworks
• Skilled in prompt engineering: crafting structured, deterministic prompts for reproducibility
• Familiarity with VSCode Extension development
• Understanding of the Model Context Protocol (MCP) or similar multi-modal context interfaces
• Web/UI development using React, HTML/CSS, or CLI-based tools
• Rapid prototyping with Gradio or Streamlit
• Knowledge of RAG pipelines, chunking strategies, and semantic retrieval using vector stores (e.g., FAISS, Chroma)
• Experience with Playwright or Selenium for web interaction/scraping
• Backend development using FastAPI or Flask
• Deployment experience with Vercel, Render, or Docker-based internal infra.

Academic Credentials:

• Graduate/master's degree in computer science or related fields

Location: Bangalore

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.