Skip to content

Migrate the Adversarial training example to the latest flax.nnx module.#1710

Open
rajasekharporeddy wants to merge 1 commit into
google-deepmind:mainfrom
rajasekharporeddy:adversarial_training
Open

Migrate the Adversarial training example to the latest flax.nnx module.#1710
rajasekharporeddy wants to merge 1 commit into
google-deepmind:mainfrom
rajasekharporeddy:adversarial_training

Migrate to flax.nnx from flax.linen in adversarial_training example

6f00efc
Select commit
Loading
Failed to load commit list.