Implementing the LeNet architecture with the MXNet library