torch Model.register_buffer()

for Masking

self.register_buffer('tril', torch.tril(torch.ones(block_size, block_size)))