v0.3.1
[lazy] support init on cuda (#4269)
[checkpoint] refactored the API and added safetensors support (#3427)