jax

1b63d5e2 - Fixed deadlock in NamedSharding ctor

Fixed deadlock in NamedSharding ctor Description: - Test timeout were seen in ColocatedPythonTest test case - GDB report: https://gist.github.com/vfdev-5/d64183f7b5dde3e666eea6cd61670128