TBH, I think I have exhausted my recommendations here 
I tend to think it’s related to the machine you’re using, in the sense that I can run this model with default init without issues on my machine. I’m afraid I’ll have to defer to others for more specifi advice however.