Module utils 

Functions

count_parameters

Calculate the number of parameters in a tensor

create_causal_mask

Create a causal mask for attention

trunc_normal_init

Truncated normal initialization