Natural Gradient implementation in Theano