optax
optax copied to clipboard
Add a projection onto L1 unit ball
Projection onto L1 unit ball is extremly useful for sparse solution. Quite surprising that is is not included in: optax/projections/_projections.py.
I would love to contribute it.