Thanks. Trying it.
By the way when i used
#trace = pm.sample(1000)
inference = pm.fit(method='advi') # this is mean field ADVI. something like 'fullrank_advi' might be tried too
trace = inference.sample(1000)
original code worked