yellowdolphin

dist_sync_fn seems dysfunctional

1

## 🐛 Bug `dist_sync_fn` is a documented kwarg of class Metric and suggests to pass an alternative to `torch.distributed.all_gather`. I suppose, the intend is to allow use of distributed contexts...

tfp.stats.histogram cannot be compiled by XLA?

1

# Summary of problem Applying tfp.stats.histogram on the data in a tf keras model breaks XLA compilation. The example code (see below) works on CPU/GPU but with TPU strategy raises:...