Update fast_gradient_method.py

cleverhans-lab / cleverhans

An adversarial example library for constructing attacks, building defenses, and benchmarking both

MIT License

6.15k stars 1.39k forks source link

Open peterljq opened 1 year ago

peterljq commented 1 year ago

The lastest Stax in JAX should be jax.example_libraries.stax as documented here: https://jax.readthedocs.io/en/latest/jax.example_libraries.stax.html