Honestly I hate this new trend of AI/ML papers not even describing the model anymore. They used a graph neural network model called M3Gnet. It's mentioned in the APPENDIX.
Nowhere in the article linked to by HN they say that, so it's easy to assume their use of ML is just marketing.