I considered that, but treating the mask as a buffer will also make it a part of the module’s state_dict which is unnecessary since the mask is just a constant. Is there a way to specify that the buffer doesn’t need to be written out to disk?