Tests

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

Job	Run time
Check the code format	1m 44s
Test the code on CPU (jax)	8m 1s
Test the code on CPU (torch)	2m 54s
Test the code on GPU (jax)	5m 54s
Test the code on TPU (tensorflow)	4m 41s
Test the code on GPU (torch)	4m 24s
Test the code on TPU (jax)	6m 18s
Test the code on CPU (tensorflow)	3m 18s
Test the code on GPU (tensorflow)	5m 27s
	42m 41s

Provide feedback