Skip to content

tf2jax 0.2.0
Compare
Choose a tag to compare
@shaobohou shaobohou released this 03 May 14:18
· 178 commits to main since this release
v0.2.0
1c5dc22

What's Changed

  • Changed required tensorflow version from nightly to 2.8.0. by @copybara-service in #8
  • Enable custom_gradient support by default. by @copybara-service in #9
  • Support NCHW in convolution, pooling and BiasAdd operations. by @copybara-service in #13 and #14
  • Fix a bug with shrink axis in StridedSlice. by @copybara-service in #17
  • Support bool in MatrixDiagV3. by @copybara-service in #16
  • Support TopKV2. by @copybara-service in #15
  • Support XlaRngBitGenerator. @copybara-service in #19

Full Changelog: v0.1.1...v0.2.0

Assets 2
Loading