Implement [`tensor_product`](https://e3nn-jax.readthedocs.io/en/latest/api/tensor_products.html#e3nn_jax.tensor_product).
Implement
tensor_product.