#747 surfaced a behavioral difference between handlers.jax.sizesof and handlers.torch.sizesof in attempting to fuse their implementation: the JAX version allows int- and Array-typed free variables, while the Torch version allows only Tensor-typed free variables. I don't think there's any deep reason for that, so we should make them fully consistent.
#747 surfaced a behavioral difference between
handlers.jax.sizesofandhandlers.torch.sizesofin attempting to fuse their implementation: the JAX version allowsint- andArray-typed free variables, while the Torch version allows onlyTensor-typed free variables. I don't think there's any deep reason for that, so we should make them fully consistent.