Add Tensorstore back to required dependencies.
Add Tensorstore back to required dependencies.

It was reversed back when user reported https://github.com/google/flax/issues/2341, but it was now fixed and verified.
google/flax,google/flax