A Pylearn2 implementation of MLPs with local winner-take-all (LWTA) blocks