google-deepmind/tf2jax v0.3.8

tf2jax 0.3.8

Repository: google-deepmind/tf2jax

Tag: v0.3.8

Published: 2025-08-21T13:18:03Z

Prerelease: no

Release notes:

What's Changed

• Drop support for python 3.10, add support for python 3.12 https://github.com/google-deepmind/tf2jax/pull/252 https://github.com/google-deepmind/tf2jax/pull/253
• Update JAX nightly index usage https://github.com/google-deepmind/tf2jax/pull/238
• Add new attr use_shardy_partitioner in _xla_call_module attr checking. https://github.com/google-deepmind/tf2jax/pull/240
• Update tests to use Flax instead of haiku https://github.com/google-deepmind/tf2jax/pull/241
• Add support for fresnel_sin and fresnel_cos https://github.com/google-deepmind/tf2jax/pull/243
• Use V2 sharding if available. https://github.com/google-deepmind/tf2jax/pull/244
• Fix TensorListGetItem tests https://github.com/google-deepmind/tf2jax/pull/251
• JAX api migration and tests updates. https://github.com/google-deepmind/tf2jax/pull/231 https://github.com/google-deepmind/tf2jax/pull/232 https://github.com/google-deepmind/tf2jax/pull/233 https://github.com/google-deepmind/tf2jax/pull/235 https://github.com/google-deepmind/tf2jax/pull/236 https://github.com/google-deepmind/tf2jax/pull/237 https://github.com/google-deepmind/tf2jax/pull/239 https://github.com/google-deepmind/tf2jax/pull/245 https://github.com/google-deepmind/tf2jax/pull/246 https://github.com/google-deepmind/tf2jax/pull/248 https://github.com/google-deepmind/tf2jax/pull/249 https://github.com/google-deepmind/tf2jax/pull/250

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