flax
297c3889 - have TrainState subclass nnx.Module
Commit
View On GitHub
Commit
2 years ago
have TrainState subclass nnx.Module
References
pr/chiamp/3782
Author
chiamp chiamp
Committer
chiamp chiamp
Parents
f234fc1b
FAQ Terms Privacy Refunds Impressum
Loading