optax icon indicating copy to clipboard operation
optax copied to clipboard

New Example on creating a custom `GradientTransformation`

Open SauravMaheshkar opened this issue 1 year ago • 2 comments
trafficstars

Motivation

Recently for a personal project (#641) I wanted to created a custom GradientTransformation object and found out that there wasn't a tutorial or walkthrough on how to create one. While the "Gradient Transformations" section in the README.md section was somewhat helpful, I still had to look around and search for example implementations in other repositories and within the contrib sub-package.

Proposal

Create a new example or a page in the documentation containing instructions on how to create a custom GradientTransformation object.


Happy to contribute and work on this.

SauravMaheshkar avatar Dec 02 '23 18:12 SauravMaheshkar

hey @SauravMaheshkar , yes, I believe a short example on this on the example/ directory would be welcomed!. I'll assign you this issue.

Thanks for suggesting this!

fabianp avatar Dec 03 '23 15:12 fabianp

Also tagging #958 and #641

SauravMaheshkar avatar May 04 '24 17:05 SauravMaheshkar