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

Avoid depending on JAX internals, which are about to change (see cl/681582933). #801

Open copybara-service[bot] opened 3 weeks ago

copybara-service[bot] commented 3 weeks ago

Avoid depending on JAX internals, which are about to change (see cl/681582933).