How about pylearn2.linear.weight_init? All of the weight initialization stuff is specific to the linear operators.