Supergrad

What's Changed
* Bump JAX version to 0.4.34 by allegro0132 in https://github.com/iqubit-org/supergrad/commit/4d465eee7ff5dd650c50d4c7b950654464bd1567
* Deprecate jax.experimental.host_callback


**Full Changelog**: https://github.com/iqubit-org/supergrad/compare/v0.2.0...v0.2.1

What's Changed
* Implement multi-GPU by sharding constraint by allegro0132 in https://github.com/iqubit-org/supergrad/pull/2


**Full Changelog**: https://github.com/iqubit-org/supergrad/compare/v0.1.4...v0.2.0

What's Changed
* Update docstring and adjust default behavior by allegro0132 in https://github.com/iqubit-org/supergrad/pull/1

New Contributors
* allegro0132 made their first contribution in https://github.com/iqubit-org/supergrad/pull/1

**Full Changelog**: https://github.com/iqubit-org/supergrad/compare/v0.1.3...v0.1.4