@ferrine Were there any updates to sample_node functionality? I remember the code below worked just fine a few months ago and I recently updated to master and I can no longer produce a working example of that kind.
data = np.random.randint(3,size=(2,2)).astype(np.float32)
with pm.Model() as model:
theta = pm.Gamma('g',1,1,shape=(2,2))
data = pm.Poisson('data',theta,observed=data)
inference = pm.ADVI()
approx = inference.fit(n=2)
@pwl This worked for me on latest master. However I have no GPU with me at the moment. You can try to set compute_test_value='ignore' as a workaround. Test values are annoying sometimes any I do not use them under the hood of VI.