API Reference

Full API documentation auto-generated from source code docstrings.

Module	Description
Dataset	Data loading: in-memory, on-disk, sharded datasets, schema
Model	GNN architectures: HeteroGNN, RGAT, HEAT, HGT, losses
Trainer	DDP training loops, multi-case training, utilities
Evaluator	Constraint violation checking, Modeler utility
Loader	Custom PyG DataLoader wrappers
Utilities	Graph construction, throughput tracking, I/O helpers