Imported from https://github.com/bryankim96/stux-DNN