Skip to content

tf2jax 0.3.4

Choose a tag to compare

View all tags
@shaobohou shaobohou released this 21 Apr 09:07
· 108 commits to main since this release
v0.3.4
f83c807

What's Changed

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

Full Changelog: v0.3.3...v0.3.4

Assets 2
Loading