A lightweight neural network simulation environment