Use sqrt(fan_in) as default in and truncated_normal

[dlwh]

better matches modern practice in the LLM space (c.f. https://www.microsoft.com/en-us/research/uploads/prod/2021/11/TP5.pdf and OLMO)