This repository recopiles all the theory on how to build neural network ansatz (Neural Quantum States from Carleo's paper) from scratch in JAX. The final goal is to find a way, based on the methodologies from Carleo, to find the excited states of such quantum many-body systems.