search

optuna / optuna-examples

Examples for https://github.com/optuna/optuna
MIT License
632 stars 171 forks source link

Flax example #221

Closed nzw0301 closed 4 months ago

nzw0301 commented 8 months ago

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.

github-actions[bot] commented 7 months ago

This issue has not seen any recent activity.

github-actions[bot] commented 4 months ago

This issue was closed automatically because it had not seen any recent activity. If you want to discuss it, you can reopen it freely.