We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
59a57b7
Bump minimum JAX version to v0.6.0 PiperOrigin-RevId: 805898990
05ea1af
Release 0.11.24. PiperOrigin-RevId: 800598967
bd9b156
Release new version. Add `construct_maximal_shardings` to public inte…
0f1588b
Release 0.11.22. PiperOrigin-RevId: 795229261
d94bad6
Release 0.11.21. PiperOrigin-RevId: 792272070
c091f04
Release 0.11.20 PiperOrigin-RevId: 788097118
f46d3f6
Release 0.11.19. PiperOrigin-RevId: 780799160
e908554
Check JAX version when using jax.layout.Format. PiperOrigin-RevId: 778111564
5028007
Increase Orbax version. PiperOrigin-RevId: 777755333
fc5eaad
No public description PiperOrigin-RevId: 773018863
