Fix weight init, deprecation warnings, safetensors (#29)

f6c1a25

verified

leoxiaobin commited on Aug 4, 2025