GraphStorm: All-in-one Graph Machine Learning Framework for Industry Applications

Graph machine learning (GML) is effective in many business applications.However, making GML easy to use and applicable to industry applications withmassive datasets remain challenging. We developed GraphStorm, which provides anend-to-end solution for scalable graph construction, graph model training andinference. GraphStorm has the following desirable properties: (a) Easy to use:it can perform graph construction and model training and inference with just asingle command; (b) Expert-friendly: GraphStorm contains many advanced GMLmodeling techniques to handle complex graph data and improve model performance;(c) Scalable: every component in GraphStorm can operate on graphs with billionsof nodes and can scale model training and inference to different hardwarewithout changing any code. GraphStorm has been used and deployed for over adozen billion-scale industry applications after its release in May 2023. It isopen-sourced in Github: https://github.com/awslabs/graphstorm.