Visualization library to aid in neural net training