Mohammad Amin Nabian

Results 138 comments of Mohammad Amin Nabian

> > @stadlmax as far as I remember, we were using fused layernorm and that gave us nice speedup (although I can't find it in the most recent code)... Did...

@stadlmax added support for TE layernorm.

> Note to myself: API updates breaks GraphCast tests. Need to update them all. Done