Pylearn2 wrappers