scico/linop/xray/_xray.py:235: jnp.zeros((len(angles), ny), dtype=im.dtype)
scico/linop/xray/_xray.py:396: proj = jnp.zeros((num_views,) + det_shape, dtype=im.dtype)
scico/linop/xray/_xray.py:441: HTy = jnp.zeros(input_shape, dtype=proj.dtype)
scico/linop/xray/_xray.py:168: unit_sino = jnp.ones(self.output_shape, dtype=np.float32)
Module
scico/linop/xray/_xray.pyhas a number of calls tojax.numpy.zerosandjax.numpy.onesthat do not allow specification of a device or a sharding.See
A branch to address this should be created by branching from branch brendt/shard.