search

google / flax

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

Remove outdated compatibility code. #4324

Closed jakevdp closed 1 month ago

jakevdp commented 1 month ago

The minimum supported JAX version is now 0.4.27, so these compatibility shims for older versions are no longer necessary.