What's Changed
- Handle bad_indices_policy for GatherNd and ScatterNd. #195
- Support more numpy ops. #197
- Support python 3.11. #201
- Drop support for python 3.9 to align with JAX. #200
- Handle bad_indices_policy for GatherNd and ScatterNd. #206
- Add the
MulNoNanop. #210 - Allow to optionally disable the type-check assert in
TensorListGetItem. #211 - Update tf2jax to use new gather/scatter batching dims. #217 #221
- Fix a bug in Slice for size==-1. #220
- Fix handling of integer types during gradient computation with mhlo modules. #225
Full Changelog: v0.3.6...v0.3.7