v0.3.0

What's Changed

• Ignore shard_map attr error in mypy. by @cantonios in #97
• Added TF specific documentation to DistributedEmbedding. by @hertschuh in #94
• Fix symbolic calls for EmbedReduce. by @hertschuh in #98
• Move DistributedEmbedding declaration to its own file. by @hertschuh in #99
• Remove dependency on tree and use keras.tree. by @hertschuh in #100
• Only enable JAX on linux_x86_64. by @cantonios in #101
• Add out_sharding argument WrappedKerasInitializer. by @cantonios in #102
• Use Python 3.10 style type annotations. by @hertschuh in #104
• Do not bundle test utils in wheel. by @hertschuh in #105
• Update version number to 0.2.1 by @hertschuh in #106
• Fix invalid escape sequence in unit test. by @hertschuh in #108
• Replace leftover unflatten_as to pack_sequence_as. by @hertschuh in #109
• Make the declaration of Nested compatible with pytype. by @hertschuh in #110
• Add ragged support for default_device placement on JAX. by @cantonios in #107
• Add documentation for using DistributedEmbedding with JAX. by @cantonios in #111
• api_gen now excludes backend specific code. by @hertschuh in #103
• Enable preprocess calls with symbolic input tensors. by @cantonios in #113
• Check for jax_tpu_embedding on JAX backend. by @cantonios in #114
• Version bump to 0.2.2.dev0 by @laxmareddyp in #117
• Fix docstring for DistributedEmbedding. by @hertschuh in #118
• Minor docstring fix in DistributedEmbedding. by @hertschuh in #120
• Fix nightly builds. by @hertschuh in #121
• Bump torch from 2.6.0 to 2.7.1 by @dependabot[bot] in #119
• Checkpoints by @silkyarora in #122
• Add FTRL and Adam optimizers. by @cantonios in #123
• Update references of jax_layout.Layout to jax_layout.Format. by @cantonios in #124
• Update explicit layouts to use JAX 0.6.2 layout.Format. by @cantonios in #125
• Change for DeviceLocalLayout in JAX 0.6.3. by @hertschuh in #126
• Fix for removal of JAX DeviceLocalLayout. by @hertschuh in #127
• Add note about multi-host dataset sharding. by @cantonios in #131
• Fix JAX Layout tiling argument for JAX 0.7.1 by @hertschuh in #133
• Updated documentation regarding DistributedEmbedding batch size. by @hertschuh in #138
• Fix for batch size of FeatureConfig.output_shape on TensorFlow. by @hertschuh in #139
• Fix handling of mixed placement in DistributedEmbedding. by @hertschuh in #136
• Fix batch sizes in distributed_embedding_test.py. by @hertschuh in #142
• Fix and clean up multihost preprocessing. by @cantonios in #146
• Bump Python, JAX and Tensorflow versions by @hertschuh in #148
• DistributedEmbedding: do not hardcode the number of SparseCores with TF. by @hertschuh in #147
• Make TableConfig and FeatureConfig unhashable. by @hertschuh in #150
• version bump to 0.3.0.dev0 by @sachinprasadhs in #151
• Update release version to 0.3.0 by @sachinprasadhs in #152

New Contributors

• @silkyarora made their first contribution in #122
• @sachinprasadhs made their first contribution in #151

Full Changelog: v0.2.1...v0.3.0