Nearly the same speed as Theano