v0.2.1

What's Changed

• Bump JAX version to 0.4.34 by @allegro0132 in 4d465ee
• Deprecate jax.experimental.host_callback

Full Changelog: v0.2.0...v0.2.1

v0.2.0

What's Changed

• Implement multi-GPU by sharding constraint by @allegro0132 in #2

Full Changelog: v0.1.4...v0.2.0

v0.1.4

What's Changed

• Update docstring and adjust default behavior by @allegro0132 in #1

New Contributors

• @allegro0132 made their first contribution in #1

Full Changelog: v0.1.3...v0.1.4

SuperGrad v0.1.3

Push initial codes