Skip to navigation

Release 3.0.0

Breaking Changes

  • Multi-node multi-GPU (MNMG) pipeline — training now requires torchrun and the new gpu: multi / gpu: single config field. Legacy non-torchrun entrypoints are no longer supported, while single-GPU runs remain supported with gpu: single (or NPROC=1).
  • New config schema — the YAML/JSON config format has changed substantially. The FullConfig schema now groups all settings under paths and models[].hyperparameters.gnn / models[].hyperparameters.xgb. Old flat configs are not compatible and must be migrated.
  • Triton server version — the serving container is now nvcr.io/nvidia/tritonserver:26.04-py3. Artifacts generated by earlier training containers are not compatible with this server version.
  • kind: GNN_XGBoost_NP — Node Prediction is now a first-class kind option in the config rather than a separate container entrypoint.
  • Mask file placement — train/val/test mask files are now placed in edges/ (EP) or nodes/ (NP). The previous location is no longer supported.
  • Edge filename parser — the src/rel/dst parsing now always takes the first token as src and the last token as dst. Multi-token node type names (e.g., credit_card) in edge filenames are no longer supported.

Release 2.0.0

Features

  • Edge Prediction
  • Builds a GNN Model
  • Train an XGBoost model to predict fraud scores
  • Uses Wholegraph and cuGraph to encapsulate graph storage and complexity efficiently
  • Support to use XGBoost modeling optionally without GNN embeddings

Release 1.0.0

Features

  • Node Prediction
  • Builds a GNN Model
  • Train an XGBoost model to predict fraud scores
  • Uses Wholegraph and cuGraph to efficiently encapsulate graph storage and complexity
  • Support to optionally us XGBoost modeling without GNN embeddings