Summary:
-
Distributed Embeddings & JAX Support:
- Improvements to the DistributedEmbedding layer (documentation, bug fixes, code structure) and the JAX backend (platform-specific enabling, new API arguments, ragged tensor support).
What's Changed
- Version bump 0.2.1.dev0 by @laxmareddyp in #115
- Version bump to 0.2.1 by @laxmareddyp in #116
New Contributors
- @laxmareddyp made their first contribution in #115
Full Changelog: v0.2.0...v0.2.1
Contributors
laxmareddyp