Gram: Recursive reasoning models with stochastic latent trajectories (10M param)

yesterday I tried to reproduce GRAM because there was no official code https://github.com/ad3002/gram On N-Queens 8x8 I get 87.58% valid solutions on one 3080Ti, paper reports 99.7%, but with much more compute