search

google-deepmind / dm-haiku

JAX-based neural network library
https://dm-haiku.readthedocs.io
Apache License 2.0
2.91k stars 231 forks source link

Make flax an optional dependency. #782

Closed copybara-service[bot] closed 6 months ago

copybara-service[bot] commented 6 months ago

Make flax an optional dependency.

Fixes #780.