feat: move model loader functionality to augmentation #496

Job	Run time
sample-config	9s
lint (framework)	3m 33s
lint (accelerated-peft)	3m 20s
lint (fused-ops-and-kernels)	3m 26s
lint (attention-and-distributed-packing)	3m 58s
lint (accelerated-moe)	3m 18s
	17m 44s

Provide feedback