Pylearn2 wrapers