Codebase for a simple Jax (using Stax) neural network for CIFAR-10.
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool