Tests

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

Job	Run time
Test the code on CPU (jax)	7m 25s
Check the code format	2m 9s
Test the code on GPU (tensorflow)	5m 29s
Test the code on GPU (torch)	4m 16s
Test the code on CPU (torch)	2m 48s
Test the code on GPU (jax)	5m 50s
Test the code on CPU (tensorflow)	3m 18s
Test the code on TPU (jax)	6m 22s
Test the code on TPU (tensorflow)	4m 42s
	42m 19s

Provide feedback