https://github.com/JosephRRB/torchgfn_spinfoams/blob/master/src/spinfoam/spinfoams.py we should reproduce their results