ptrblck:

lstm = nn.LSTM(1, 1)
torch.nn.init.xavier_uniform_(lstm.weight_ih_l0)



Thanks this works, not sure about its impact on training yet but the reference to  .weight_ih_l0 is what I was looking for