adding a perturbations module. Can take pytrees as inputs - Githubissues

google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.

https://optax.readthedocs.io

Apache License 2.0

1.63k stars 174 forks source link

adding a perturbations module. Can take pytrees as inputs #827

Open copybara-service[bot] opened 6 months ago

copybara-service[bot] commented 6 months ago

adding a perturbations module. Can take pytrees as inputs