Role description

Knowledge Graph Engineer

Specialist I - Product Engineering

UST is searching for an experienced Knowledge Graph Engineer to design, build, and scale a production-grade property graph platform that powers customer segmentation, device intelligence, and household-level insights.

The opportunity:
• Schema Design: Architect the property graph schema defining node types
• Bulk Load Pipeline: Build and validate the initial bulk load job across the ingestion stack (e.g., Delta Lake → S3 staging → Neptune bulk loader or equivalent technology).
• Real-Time CDC Pipeline: Implement a change data capture pipeline (e.g., Cosmos DB Change Feed → Kafka → Flink → Neptune writer) with an end-to-end lag target of • Query Development: Write and optimize Gremlin traversal queries for household segmentation, device-sharing patterns, and account-linked segmentation use cases.
• Index Strategy: Design vertex-centric indexes and leverage Neptune Analytics HNSW for embedding-based similarity lookups.
• Graph Embeddings: Build a Node2Vec embedding pipeline (SageMaker or Databricks) and load SIMILAR_TO edges to support ML-driven similarity features.
• Documentation: Document schema definitions, traversal patterns, and query performance benchmarks for consumption by the segment engine team

This position description identifies the responsibilities and tasks typically associated with the performance of the position. Other relevant essential functions may be required.

What you need:
• Skills: Graph DB (Neptune OR Neo4j OR TigerGraph) - Gremlin OR Cypher
• Good to have: - Neptune specific - Flink - Embeddings
• 4+ years of graph database engineering experience with production Gremlin / TinkerPop expertise.
• AWS Neptune or equivalent cloud graph database - bulk loader operations, instance sizing, HA configuration, and VPC networking.
• Apache Kafka and Apache Flink for CDC pipeline design and implementation.
• Property graph data modelling - entity resolution, edge cardinality, and partition key design.
• Graph traversal performance profiling at scale (100M+ nodes).
• Nice-to-Have Skills
• Graph embedding algorithms - Node2Vec, GraphSAGE, or similar.
• Neptune Analytics experience for graph analytics workloads.
• Neo4j migration or comparative architecture experience (trade-offs vs. Neptune at scale).
• Python (gremlinpython) and Java traversal source authoring.
• AWS SageMaker or Azure Databricks for embedding model training.

Compensation can differ depending on factors including but not limited to the specific office location, role, skill set, education, and level of experience. UST provides a reasonable range of compensation for roles that may be hired in various U.S. markets as set forth below.

Role Location: Remote-US

Compensation Range: $154,000-$231,000

Benefits

Full-time, regular employees accrue a minimum of 10 days of paid vacation per year, receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year), 10 paid holidays, and are eligible for paid bereavement leave and jury duty. They are eligible to participate in the Company's 401(k) Retirement Plan with employer matching. They and their dependents residing in the US are eligible for medical, dental, and vision insurance, as well as the following Company-paid Employee Only benefits: basic life insurance, accidental death and disability insurance, and short- and long-term disability benefits. Regular employees may purchase additional voluntary short-term disability benefits, and participate in a Health Savings Account (HSA) as well as a Flexible Spending Account (FSA) for healthcare, dependent child care, and/or commuting expenses as allowable under IRS guidelines. Benefits offerings vary in Puerto Rico.

Part-time employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year) and are eligible to participate in the Company's 401(k) Retirement Plan with employer matching.

Full-time temporary employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year) and are eligible to participate in the Company's 401(k) program with employer matching. They and their dependents residing in the US are eligible for medical, dental, and vision insurance.

Part-time temporary employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year).

All US employees who work in a state or locality with more generous paid sick leave benefits than specified here will receive the benefit of those sick leave laws.

#UST

#CB

#LI-VK3

#LI-Remote