Migrate the Adversarial training example to the latest flax.nnx module.#1710
Open
rajasekharporeddy wants to merge 1 commit into
Open
Migrate the Adversarial training example to the latest flax.nnx module.#1710rajasekharporeddy wants to merge 1 commit into
Adversarial training example to the latest flax.nnx module.#1710rajasekharporeddy wants to merge 1 commit into
