search

google-research / jax3d

Apache License 2.0
729 stars 94 forks source link

[JAX] Fix pytype errors from an upcoming change that improves JAX typing inference under pytype #166

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

[JAX] Fix pytype errors from an upcoming change that improves JAX typing inference under pytype