Can MULTILEVEL MODELING example be run on GPU?

[Bolor-Erdene Zolbayar]

Hello, 

I was wondering if tensorflow MULTILEVEL MODELING can be run on GPU. I am having some scalability issue with PyMC3. Do you guys have training accelerator with the MULTILEVEL modeling? 

Best,

Bolor

[Bolor-Erdene Zolbayar]

I am also trying to create a docker image for your example. Do you guys have a docker image prepared for your example https://www.tensorflow.org/probability/examples/Multilevel_Modeling_Primer ? 

Your help would be much appreciated. Thanks !

Bolor