Code for a publication for Nature Communications AI & Computing - View it on GitHub