Release v0.2.4

PiperOrigin-RevId: 695823027

Release v0.2.3

PiperOrigin-RevId: 650608435

Release of version 0.2.2

PiperOrigin-RevId: 619580416

0.2.1 release

PiperOrigin-RevId: 613637111

v0.2.0 release

PiperOrigin-RevId: 612851804

v0.1.9 release

PiperOrigin-RevId: 603973640

New optax release

PiperOrigin-RevId: 598600922

Remove deprecated zip_safe field, and potentially fix the deploy work…

…flow

PiperOrigin-RevId: 551313330

Add optimizer code for Mechanic learning rate tuner (https://arxiv.or…

…g/pdf/2306.00144.pdf) to Optax.

PiperOrigin-RevId: 550679862

Release v0.1.5.

PiperOrigin-RevId: 525743135