Tests

Revert "Revert 'Enables the huggingface checkpoint conversion to MaxText orbax.'" #1274

Job	Run time
prelim	1s
gpu_image / Build and upload image (a100-40gb-4)	43s
tpu_image / Build and upload image (v4-8)	1m 46s
gpu_unit_tests / run	3m 11s
gpu_integration_tests / run	7m 35s
tpu_integration_tests / run	6m 14s
tpu_unit_tests / run	19m 3s
Clean up	9s
Notify failed build	2s
	38m 44s

Provide feedback