Add comprehensive causal and custom masking support

Implements attention masking with: - AttentionMaskType enum (.none, .causal, .custom) - Metal kernel-level masking - Integration across forward/backward passes for gradient correctness - Tests & docs