Bump up the version to v0.1.0

Merge pull request google#442 from google:v0.1

PiperOrigin-RevId: 813117769

Adds prod and dev mode for tunix.

Merge pull request google#372 from google/lance-flax

Use Flax release insteading of github head.

tunix v0.0.1 release