chore: Move dcp to safetensors to a wrapper function for downstream usage #562

Job	Run time
sample-config	7s
lint (framework)	3m 5s
lint (accelerated-peft)	3m 41s
lint (fused-ops-and-kernels)	3m 17s
lint (attention-and-distributed-packing)	4m 1s
lint (accelerated-moe)	3m 15s
	17m 26s

Provide feedback