NeuralSolvers
NeuralSolvers copied to clipboard
Modularize fit function
Currently, the fit function is a very big function that includes tracking, fitting, parallelization, etc. Here we should modularize the functionalities to get a more maintainable code.