@override List<Tensor> parameters() => [ ...attention.parameters(), ...ffn.parameters(), ...ln1.parameters(), ...ln2.parameters(), ];