Skip to content

tf2jax 0.3.5

Choose a tag to compare

@shaobohou shaobohou released this 28 Jul 13:23
· 89 commits to main since this release

What's Changed

  • Better error message for missing inputs in #131
  • Support sharding annotation in #132, #133
  • Support shape refinement with XlaCallModule in #140, in #144, in #145
  • Add LowerBound and UpperBound operation to tf2jax to support searchsorted. in #141
  • Dropped support for python 3.8 in #140,
  • Handle shape check asserts in in #146

Full Changelog: v0.3.4...v0.3.5