Does TVM currently support TensorFlow Probability (GitHub - tensorflow/probability: Probabilistic reasoning and statistical analysis in TensorFlow), specifically tfp.layers (probability/tensorflow_probability/python/layers at main · tensorflow/probability · GitHub)?
I am working with Bayesian Neural Network models (probability/bayesian_neural_network.py at main · tensorflow/probability · GitHub) 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.