flax
2d6aa629 - [nnx] revive TrainState toy example

Commit

1 year ago

[nnx] revive TrainState toy example

References

#4226 - [nnx] revive TrainState toy example

Author

cgarciae

cgarciae

Parents

Loading