Hi,
The difference I see between the two is that the one with Linear layers include some bias while the other one does not.
Also when you say it works for float32. It is only for that random seed I guess no?