flax

5a2da6d2 - Copybara import of the project:

Copybara import of the project: -- 6622efcf2782de71b89e62b2943f526c82bd2f74 by vfdev-5 <vfdev.5@gmail.com>: 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 PiperOrigin-RevId: 822202664