Remove CompiledFunction from Haiku since the default jax version >= 0.4.6 - Githubissues

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

Remove CompiledFunction from Haiku since the default jax version >= 0.4.6 #628

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

Remove CompiledFunction from Haiku since the default jax version >= 0.4.6