Machine Learning Engineer, AGI - Amazon Knowledge Graph

Amazon's AGI Information is seeking an exceptional Software Development Engineer to drive advancements in the Amazon Knowledge Graph (AKG) Construction and Retrieval infrastructure. AKG is re-inventing knowledge graphs for the LLM-era, optimizing for LLM grounding and enabling agentic AI systems that require real-time knowledge retrieval. AKG serves as the foundational knowledge layer for Nova and Alexa+, providing structured, factual information that grounds LLM responses and powers autonomous agent decision-making across Amazon surfaces.
As a member of the AKG Interfaces team, you will have the opportunity to work on interesting problems with immediate customer impact. The team is addressing challenges in automating gap detection, data ingestion, fact verification, query optimization, entity resolution, and building scalable APIs that serve billions of requests from agentic systems.

Key job responsibilities
Build a scalable information grounding infrastructure that provides grounding evidence to multiple Amazonian experiences and agentic AI systems. Grounding large language models (LLMs) and AI agents ensures that their interactions with users are informed, relevant, consistent, and trustworthy. It ensures accuracy and relevance by providing context-specific responses and mitigating the risk of generating false information through techniques like Retrieval-Augmented Generation (RAG).

Design & Write code: Write code primarily in Java and design scalable, fault tolerant and distributed applications running on AWS technology. Leverage state of the art AWS resources in the development of a scalable web grounding service for LLM based applications.

Troubleshoot and fix operational problems: You will work with systems engineers to troubleshoot operational problems and fix them. If you are so inclined, you will also automate troubleshooting procedures and write tools. And when the issues concern specific customers, you will also have the opportunity to talk to them and understand their space better.

Influence product direction: You will propose your ideas for the future of the space to product
management and get to shape the roadmap.

Hire and mentor others: You will get to interview people for the team and to mentor other engineers.

Basic Qualifications

- 3+ years of non-internship professional software development experience
- 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- Experience programming with at least one software programming language
- 4+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience

Preferred Qualifications

- 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
- Bachelor's degree in computer science or equivalent
- Experience working with ML frameworks such as Pytorch, TensorRT, Onnx, AWS Neuron and accelerating deep learning models for GPU/CPU/Inferentia architectures.

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Los Angeles County applicants: Job duties for this position include: work safely and cooperatively with other employees, supervisors, and staff; adhere to standards of excellence despite stressful conditions; communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service; and follow all federal, state, and local laws and Company policies. Criminal history may have a direct, adverse, and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above, as well as the abilities to adhere to company policies, exercise sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation. Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

Our compensation reflects the cost of labor across several US geographic ma