@override List<Value> parameters() { return [ ...attention.parameters(), ...ffn.parameters(), ...ln1.parameters(), ...ln2.parameters(), ]; }