Tests

Migrate Gpt3 to NNX. #11041

Job	Run time
prelim	2s
gpu_image / Build and upload image (a100-40gb-4)	7m 56s
tpu_image / Build and upload image (v4-8)	5m 43s
tpu_unit_tests / run	19m 5s
tpu_pathways_unit_tests / run	17m 35s
tpu_integration_tests / run	13m 1s
cpu_unit_tests (1) / run	7m 58s
cpu_unit_tests (2) / run	17m 32s
cpu_unit_tests (3) / run	18m 3s
cpu_unit_tests (4) / run	10m 10s
gpu_integration_tests / run	20m 43s
gpu_unit_tests / run	11m 34s
Clean up	7s
Notify failed build	3s
Close issue after 3 successful builds	0s
	2h 29m 32s

Provide feedback