Modelling Structured Data Learning with Restricted Boltzmann Machines in the Teacher-Student Setting

First release of the code used to reproduce the figures of the paper https://arxiv.org/abs/2410.16150. Tested on python 3.8.5 with numpy 1.24.3, matplotlib 3.7.5, scipy 1.10.1, cmasher 1.6.3, jax 0.4.6 and torch 2.2.0+cu118.