Nice, it seems to be pretty light weight as well - would you be interested in porting it upstream to theano-pymc?