NaN occurred in optimization at first Iteration with ADVI

Could you try lower the initial learning rate?