What's Changed
- Add Cumprod to tf2jax ops. #149
- Fix squeeze in tf2jax for empty tuple axis arguments. #150
- Update expected attrs of VarHandleOp -- add debug_name. #154
- Only enable shape assertions during refinement if all input shapes are static. #155
- Refactor is_poly_dim to avoid depending on hard-coded module path. #156
- Avoid reuse of polymorphic variable names in tf2jax internal. #158
- Update deepmind -> google-deepmind. #159
- Change conversion of library functions and gradient functions to be lazy. #162
- Fix gradient function lookup error when higher order gradient functions are referenced in a jax2tf model but not serialized. #164
- Enforce autograph=False on gradient function. #172
- Update chex and haiku version in #175
- Fix shape inference for jax2tf vjp functions. #180
- Support native serialization with multiple target platforms. #183
Full Changelog: v0.3.5...v0.3.6