Add support for `axlearn` #3966

Job	Run time
metadata	1s
bump-manifest	14s
arm64 / build-base / build-base	3m 37s
amd64 / build-base / build-base	3m 9s
arm64 / build-jax / build-jax	23m 34s
amd64 / build-jax / build-jax	22m 26s
amd64 / build-axlearn / build-axlearn	12m 17s
arm64 / build-axlearn / build-axlearn	7m 18s
arm64 / collect-docker-tags	1s
arm64 / test-axlearn-eks	0s
arm64 / test-axlearn-fuji-models-eks	0s
amd64 / test-axlearn-fuji-models-eks	20m 48s
amd64 / test-axlearn-eks	1h 39m 24s
amd64 / collect-docker-tags	1s
make-publish-configs	1s
merge-new-manifest	0s
finalize / workflow-badge	3s
finalize / report	4s
finalize / upload-badge	4s
finalize / publish-badge	5s
	3h 13m 7s

Provide feedback