Toolkit to train neural networks