Taro Logo

Software Development Engineer (Data/ML), AGI Information - Knowledge Graphs

A global technology company leading in e-commerce, cloud computing, AI, and digital streaming.
$129,300 - $223,600
Machine Learning
Mid-Level Software Engineer
In-Person
5,000+ Employees
3+ years of experience
AI · Enterprise SaaS

Job Description

The Amazon Knowledge Graph (AKG) team is pioneering the next generation of knowledge graphs for the LLM-era. This role combines software engineering with machine learning to build sophisticated ML models and pipelines that enable efficient LLM grounding and power LLM-based customer experiences.

As a Software Development Engineer on the team, you'll be responsible for:

  • Developing efficient data processing pipelines for large-scale training and inference
  • Building and maintaining experimentation and A/B testing infrastructure
  • Creating LLM-assisted tools for knowledge graph creation, including automated ontology generation and real-time fact extraction
  • Architecting AI/ML systems that power Amazon's billion-entity knowledge graph
  • Implementing ML model infrastructure for high-throughput, low-latency production inference
  • Collaborating with scientists to productionize ML models and develop new knowledge mining architectures

You'll work at the intersection of big data, machine learning, and distributed systems, applying professional software engineering practices across the full development lifecycle. This is an opportunity to shape how Amazon organizes and leverages knowledge at unprecedented scale.

The role offers competitive compensation ($129,300-$223,600 base salary depending on location) plus equity and comprehensive benefits. You'll be part of a team pushing the boundaries of knowledge representation in the age of large language models, with the resources and scale of Amazon behind you.

Last updated 2 days ago

Responsibilities For Software Development Engineer (Data/ML), AGI Information - Knowledge Graphs

  • Develop efficient data processing pipelines for large-scale training and inference
  • Support experimentation and A/B testing infrastructure
  • Participate in code reviews and technical design discussions
  • Develop LLM-assisted tools for knowledge graph creation
  • Architect AI/ML systems for billion-entity knowledge graph
  • Release and maintain ML model infrastructure
  • Collaborate with applied scientists to productionize ML models

Requirements For Software Development Engineer (Data/ML), AGI Information - Knowledge Graphs

Python
Java
Kafka
  • 3+ years of non-internship professional software development experience
  • Bachelor's degree in computer science or equivalent
  • 2+ years of non-internship design or architecture experience
  • Knowledge of professional software engineering & best practices
  • Experience with data processing and ETL pipelines at scale (Spark, GlueJob, Kafka)

Benefits For Software Development Engineer (Data/ML), AGI Information - Knowledge Graphs

Medical Insurance
Equity
401k
  • Medical Insurance
  • Equity
  • 401k