Squared error loss
It would be great if there was an implementation of a squared error loss function, because it is a very basic loss function and the implementation would be a good starting point for implementing other custom losses. It's probably not too hard to implement, but currently the complexity of the softmax cross entropy implementation makes it hard to figure out how to implement other losses.