flax

5892e2bc - added attention refactor to changelog