Right sorry about the float32. Let’s ignore it and I have edited the question.
I have changed part of the code into:
model = torch.nn.Sequential(
    torch.nn.Linear(D_in, H, bias=False),
    torch.nn.ReLU(),
    torch.nn.Linear(H, D_out, bias=False)
)

But still get different result.
497 802.9330597078987
498 802.4759949553443
499 802.0193235987608

Is there anything that I still did wrong here? Thanks