Added theano support for AE