[WIP] Initial PR for generating and loading state dict #3228

Job	Run time
compile-gguf (macos-14)	4m 1s
gather-models-cpu	11s
gather-models-gpu	14s
runner-aoti (16-core-ubuntu)	2m 23s
runner-aoti (macos-14-xlarge)	2m 5s
runner-et (16-core-ubuntu)	6m 8s
runner-et (macos-14-xlarge)	6m 14s
test-gguf-util (macos-14)	4m 15s
test-mps (macos-m1-stable)	2m 10s
test-mps-dtype (macos-m1-stable)	2m 47s
test-tinystories-executorch (16-core-ubuntu)	10m 17s
test-tinystories-executorch (macos-14-xlarge)	9m 59s
test-torchao-experimental (macos-14-xlarge)	12m 12s
torchchat-command-load-test (macos-14)	1m 40s
test-build-runner-et-android / linux-job	14m 1s
test-cpu-eval-sanity-check (x86_64, stories15M)	10m 59s
test-cpu-eval-sanity-check (aarch64, stories15M)	10m 9s
test-cpu-compile (x86_64, stories15M)	16m 55s
test-cpu-eval-sanity-check-float16 (x86_64, stories15M)	30m 12s
test-cpu-compile (aarch64, stories15M)	9m 29s
test-cpu-aoti (x86_64, stories15M)	2m 32s
test-cpu-eval-sanity-check-float16 (aarch64, stories15M)	6m 25s
test-cpu-aoti (aarch64, stories15M)	1m 24s
test-cpu-eval-sanity-check-float32 (x86_64, stories15M)	38m 30s
test-cpu-eval-sanity-check-float32 (aarch64, stories15M)	8m 11s
test-gpu-compile (cuda, stories15M) / linux-job	24m 38s
test-gpu-aoti-bfloat16 (cuda, stories15M) / linux-job	8m 3s
test-gpu-aoti-float16 (cuda, stories15M) / linux-job	8m 2s
test-gpu-aoti-float32 (cuda, stories15M) / linux-job	8m 3s
test-gpu-eval-sanity-check (cuda, stories15M) / linux-job	12m 17s
	4h 34m 26s

Provide feedback