[JAX] Suppress pytype errors caused by the removal of ShardedDeviceArray from t5x.

google-research / pix2struct

Apache License 2.0

598 stars 54 forks source link

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

[JAX] Suppress pytype errors caused by the removal of ShardedDeviceArray from t5x.