napistu_torch.lightning

Lightning integration for napistu-torch.

This subpackage provides integration with PyTorch Lightning for training and evaluating models.

Modules

callbacks	Custom Lightning callbacks for Napistu-Torch.
constants	Lightning-specific constants.
datamodule	Base Lightning DataModule for Napistu networks.
edge_batch_datamodule	DataModule for mini-batch training on edge prediction tasks.
full_graph_datamodule	DataModule for full-batch training on a single graph.
tasks	Lightning task adapters which handle configuration and train/val/test/predict logic.
trainer	Config-aware Trainer for Napistu-Torch.