Skip to content

Feature Quantization [5/5]: Distributed Integration - #737

Draft
jchmura-sc wants to merge 1 commit into
jchmura/feature_quant_read_tfrecordsfrom
jchmura/feature_quant_integration
Draft

Feature Quantization [5/5]: Distributed Integration#737
jchmura-sc wants to merge 1 commit into
jchmura/feature_quant_read_tfrecordsfrom
jchmura/feature_quant_integration

Conversation

@jchmura-sc

@jchmura-sc jchmura-sc commented Aug 4, 2026

Copy link
Copy Markdown
Collaborator

Purpose
The purpose of this PR is to integrate feature quantization into our distributed data, loading, partioning, and sampling setup:

  • graph partitioner partitions quantized node feature using node partion book
  • neighbor samplers fetch quantized node features over rpc during subgraph hydration
  • dataloaders dequantize/reconstruct node feature during collate_fn on gpu
  • distributed dataset takes quantization metadata (to pass down to dataloaders) and new quantized node feature store

Context

Updated Changelog.md? NO

Ready for code review?: NO

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant