google-deepmind/tf2jax v0.3.4

tf2jax 0.3.4

Repository: google-deepmind/tf2jax

Tag: v0.3.4

Published: 2023-04-21T09:07:20Z

Prerelease: no

Release notes:

What's Changed

• Add documentation regarding jax2tf native serialization and XlaCallModule. https://github.com/deepmind/tf2jax/pull/123
• Improving support for XlaCallModule. https://github.com/deepmind/tf2jax/pull/97, https://github.com/deepmind/tf2jax/pull/122, https://github.com/deepmind/tf2jax/pull/118, https://github.com/deepmind/tf2jax/pull/100
• Support FFT, FFT2D, FFT3D, IFFT, IFFT2D, IFFT3D, RFFT2D, RFFT3D, IRFFT2D, IRFFT3D. https://github.com/deepmind/tf2jax/pull/115
• Support Bessel0e, Bessel1e and PopulationCount. https://github.com/deepmind/tf2jax/pull/114
• Support LeakyReluOp. https://github.com/deepmind/tf2jax/pull/120
• Support BatchToSpaceND, SpaceToBatchND. by https://github.com/deepmind/tf2jax/pull/126
• Support MatrixSetDiagV3. https://github.com/deepmind/tf2jax/pull/111
• Handle VarHandleOp . https://github.com/deepmind/tf2jax/pull/105
• Add placeholder for XlaSharding. https://github.com/deepmind/tf2jax/pull/103
• Better error message for missing parameters. https://github.com/deepmind/tf2jax/pull/110
• Better handling of static arguments for ScatterND. https://github.com/deepmind/tf2jax/pull/112
• Improved handling of non-tensor outputs. https://github.com/deepmind/tf2jax/pull/98

Full Changelog: https://github.com/deepmind/tf2jax/compare/v0.3.3...v0.3.4