Yes, if you’re up for the challenge. You could implement the Op and its gradient in theano and then use that.