Flax example
nzw0301 opened this issue · 2 comments
Motivation
Flax is a JAX-based neural network library. I'm not familiar with this, but this is a recommended replacement of dm-haiku
(see Important
section in Haiku's README
file).
So Optuna example with Flax would be beneficial for the flax/JAX community.
Description
Add flax examples like https://github.com/optuna/optuna-examples/tree/main/haiku
Alternatives (optional)
Additional context (optional)
dm-haiku repo announces
Haiku will remain best-effort supported, however the project will enter maintenance mode, meaning that development efforts will be focussed on bug fixes and compatibility with new releases of JAX.
New releases will be made to keep Haiku working with newer versions of Python and JAX, however we will not be adding (or accepting PRs for) new features.
This issue has not seen any recent activity.