Oh, nvm
I made a typing error. Instead of doing
y_pred=why.mm(ht)

i by mistake typed this and it never came to my attention
y_pred=why.mm(ht_1)

altough i  am stil not sure why this gave me nonetype grad