📚 Documentation

It likely should be mentioned that the attn_mask argument of MHA is an additive mask (-inf masks values), rather than the standard multiplicative mask (0 masks values). Perhaps even enforce a value check (all values should be 0 / -inf ?, otherwise print warning?)

cc @brianjo @mruberry @albanD @jbschlosser @zhangguanheng66