google-deepmind/tf2jax v0.3.7

tf2jax 0.3.7

Repository: google-deepmind/tf2jax

Tag: v0.3.7

Published: 2025-01-08T18:16:50Z

Prerelease: no

Release notes:

What's Changed

• Handle bad_indices_policy for GatherNd and ScatterNd. https://github.com/google-deepmind/tf2jax/pull/195
• Support more numpy ops. https://github.com/google-deepmind/tf2jax/pull/197
• Support python 3.11. https://github.com/google-deepmind/tf2jax/pull/201
• Drop support for python 3.9 to align with JAX. https://github.com/google-deepmind/tf2jax/pull/200
• Handle bad_indices_policy for GatherNd and ScatterNd. https://github.com/google-deepmind/tf2jax/pull/206
• Add the MulNoNan op. https://github.com/google-deepmind/tf2jax/pull/210
• Allow to optionally disable the type-check assert in TensorListGetItem. https://github.com/google-deepmind/tf2jax/pull/211
• Update tf2jax to use new gather/scatter batching dims. https://github.com/google-deepmind/tf2jax/pull/217 https://github.com/google-deepmind/tf2jax/pull/221
• Fix a bug in Slice for size==-1. https://github.com/google-deepmind/tf2jax/pull/220
• Fix handling of integer types during gradient computation with mhlo modules. https://github.com/google-deepmind/tf2jax/pull/225

Full Changelog: https://github.com/google-deepmind/tf2jax/compare/v0.3.6...v0.3.7