TODO: theano revisited