Changelog¶
All notable releases of lumina-sdk are documented here.
The format follows Keep a Changelog, and the project adheres to PEP 440.
[v0.1.0rc1] — 2026-04-30¶
First public release candidate.
Added¶
- Datasets —
OPFDataset(in-memory),OPFOnDiskDataset(SQLite/RocksDB),OPFShardedIterableDataset, plus ato_float32PyG transform for fp64 → fp32 casting. - Models — Heterogeneous GNN architectures:
OPFHeteroGNN(withsage/gcn/gin/gatbackends),RGAT,HEAT,HGT; homogeneous variants underOPFHomoGNN. - Losses —
OPFLossManagerwith MSE / RMSE / MAE / MAPE / SmoothL1;PhysicsInformedLossskeleton with quadratic / absolute / log-barrier penalties. - Trainer — Plain PyTorch DDP via MPI; gradient clipping; non-finite loss handling; W&B integration; sample-based scheduling.
- Evaluator —
ACOPFConstraintEvaluatorfor voltage / generation / power-balance / thermal-limit violations;Modelerfor checkpoint loading and end-to-end prediction. - Examples —
train_opf_simple.py(single-process smoke test),train_opf_ddp.py(multi-GPU DDP),evaluate_opf_constraint.py(constraint metrics). - Backends — Tested on CPU and NVIDIA CUDA; HPC job templates for Polaris (PBS / MPICH) and Perlmutter (SLURM).
- Docs — Material for MkDocs site at https://argonne-gridfm.github.io/lumina-sdk/ with quickstart, training tutorial, evaluation, multi-case, HPC, and 6 runnable Jupyter notebooks.