add ShardedDeviceArray to ad vspace op handlers
add ShardedDeviceArray to ad vspace op handlers

fixes #2529 (thanks, @dpfau !)

google/jax,google/jax,tensorflow/probability,google/jax,tensorflow/probability,google/jax