Getting Apple's Tensorflow implementation to play nicely with TFP

Hi all, anybody have any luck getting TensorFlow probability to play nicely with Apples M1 implementation? I have tried various ways to install and build but can't seem to get it to work correctly. Installing through pip gives me a runtime error of UNIMPLEMENTED: Could not find compiler for platform METAL. Building from source using bazel gives the same result. Using conda I get AttributeError: module 'tensorflow_probability.python.sts' has no attribute 'build_factored_surrogate_posterior which could be that this version is 0.17.0 and not 0.18.0.

Thanks.

Getting Apple's Tensorflow implementation to play nicely with TFP
 
 
Q