Add support for `axlearn` #3915

Job	Run time
metadata	1s
bump-manifest	13s
arm64 / build-base / build-base	3m 35s
amd64 / build-base / build-base	3m 16s
arm64 / build-jax / build-jax	19m 9s
amd64 / build-jax / build-jax	18m 49s
arm64 / build-axlearn	1s
arm64 / test-axlearn-fuji-models-eks	1s
arm64 / test-axlearn-eks	1s
arm64 / collect-docker-tags	1s
amd64 / build-axlearn	1s
amd64 / test-axlearn-eks	1s
amd64 / collect-docker-tags	1s
amd64 / test-axlearn-fuji-models-eks	1s
merge-new-manifest	1s
make-publish-configs	1s
publish-containers	1s
finalize	1s
	45m 15s

Provide feedback