Add annotations to QKV layers #291
Description
Related to elixir-nx/axon#549
LoRA targets QKV nodes. Right now the only way to do it is by inferring based on layer name. We can provide more information by giving specific metadata (such as op_name) for which layers are QKV