Tests

Add support for `DistributedEmbedding` for Ironwood and expose `has_sparsecores`. #1047

Job	Run time
Test the code on TPU (tensorflow)	4m 33s
Check the code format	2m 8s
Test the code on CPU (tensorflow)	3m 37s
Test the code on GPU (torch)	4m 23s
Test the code on GPU (tensorflow)	5m 30s
Test the code on TPU (jax)	6m 19s
Test the code on CPU (jax)	7m 9s
Test the code on GPU (jax)	5m 44s
Test the code on CPU (torch)	2m 44s
	42m 7s

Provide feedback