Guard host transfers inside pure_callbacks from deadlocking the TPU.
Guard host transfers inside pure_callbacks from deadlocking the TPU.

Also fix python/callback.cc to not swallow errors in numpy conversions.

Reverts changelist 506335783
