The AGI Amazon Knowledge Graph team develops the services and data layers that comprise a large Graph Database, providing an LLM-optimized foundational component to the AGI tech stack. The Data Vending team is responsible for designing and developing the services and infrastructure that vend data from knowledge graphs with high throughput and low latency.

Key job responsibilities
As a Machine Learning Engineer on the AKG Data Vending team, you will be a core contributor to the services and infrastructure that power Amazon’s LLM experiences. You will own the design and development of new ML features and help to improve retrieval and ranking from our data indexes