search

google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.
https://optax.readthedocs.io
Apache License 2.0
1.65k stars 181 forks source link

Test on Python 3.12 #862

Open fabianp opened 6 months ago

fabianp commented 6 months ago

seems that is pytype now that isn't compatible with python 3.12 ...