Does TVM currently support TensorFlow Probability (https://github.com/tensorflow/probability), specifically tfp.layers (https://github.com/tensorflow/probability/tree/main/tensorflow_probability/python/layers)?
I am working with Bayesian Neural Network models (https://github.com/tensorflow/probability/blob/main/tensorflow_probability/examples/bayesian_neural_network.py) and would love to use TVM for the same. If not TensorFlow Probability, then does TVM support Bayesian Neural networks from other frameworks (MxNet, Pytorch etc.)? My specific doubt is related to importing pre-trained Bayesian models into TVM where weights are not fixed point values but rather probability distributions. Thanks for your help on this. --- [Visit Topic](https://discuss.tvm.apache.org/t/tvm-support-for-tensorflow-probability/10379/1) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/2f7bef4f0fcb67d53d029bc85dc4d3bfb1ce3073c038ee3885a88b867de464c0).