Mass matrix contains zeros on the diagonal

Looks like a problem related to Google Colab then :thinking: Sorry but I can’t help you here.
Can you be more specific about what solved your problem please: was it specifying adapt_diag or downgrading to PyMC 3.8?