flax

38d0dd13 - 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.