tf2jax 0.3.5
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