What's Changed
- Update error descriptions for when result sharding is unlikely to be as expected. by @copybara-service in #13
- Fix incorrect explanation of whether sharding constraints are applied to the result of a broadcast. by @copybara-service in #14
- Adds mesh to broadcast in drjax API. by @copybara-service in #15
- Allow users to specify placement names and sizes directly with kwargs of
drjax.program. by @copybara-service in #16 - Removed Jax version limit for DrJAX. by @copybara-service in #17
- Increment the version to 0.1.4. by @copybara-service in #18
Full Changelog: v0.1.3...v0.1.4