Framework for working with neural network