v0.1.4

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

v0.1.3

What's Changed

• Update absl-py dependency in the pip package building config. by @copybara-service in #12

Full Changelog: v0.1.2...v0.1.3

v0.1.2

What's Changed

• Change ~= to >= for absl-py and chex dependencies in #10

Full Changelog: v0.1.1...v0.1.2

v0.1.1

What's Changed

• Update version for release 0.1.1. by @copybara-service in #7
• Remove use of RNG-derived outputs in tests, to ensure compatibility across JAX versions. by @copybara-service in #8
• Remove RNG usage in impls_test. by @copybara-service in #9
• Update DrJax to be Jax 0.5.x compatible by in #6

Full Changelog: v0.1.0...v0.1.1

v0.1.0

What's Changed

• Add missing __init__.py file. by @copybara-service in #2
• Remove TF dependencies (only used in testing) and replcase with chex. by @copybara-service in #3
• Add GitHub workflow to test the project. by @copybara-service in #1
• Update Bibtex citation, and remove outdated installation instructions. by @copybara-service in #4

New Contributors

• @copybara-service made their first contribution in #2

Full Changelog: https://github.com/google/drjax/commits/v0.1.0