flax
a294f849 - have TrainState subclass nnx.Module
Commit
View On GitHub
Commit
1 year ago
have TrainState subclass nnx.Module
Author
chiamp chiamp
Committer
cgarciae cgarciae
Parents
f234fc1b
FAQ Terms Privacy Refunds Impressum
Loading