Support for python 3.14
Adapted the way to access `__annotations__` object.
Flax linen code relies on `cls.__dict__["__annotations__"]`, but
in Python 3.14+ `cls.__dict__` does not contain `__annotation__` key anymore.
Fixes https://github.com/google/flax/issues/5027
