search

google / flax

Flax is a neural network library for JAX that is designed for flexibility.
https://flax.readthedocs.io
Apache License 2.0
5.78k stars 610 forks source link

upgrade python to 3.10 + use pyupgrade #4038

Closed cgarciae closed 5 days ago

cgarciae commented 6 days ago

What does this PR do?

Upgrades to python 3.10 and add pyupgrade as a pre-commit hook.