Skip to content

tf2jax 0.3.2

Compare
Choose a tag to compare
@shaobohou shaobohou released this 23 Jan 11:56
· 119 commits to main since this release

What's Changed

  • Support XlaOptimizationBarrier. #63
  • Support TensorListFromTensor, TensorListGetItem, TensorListSetItem, TensorListReserve, TensorListStack. #78, #79
  • Support DvNoNan, ResourceGather, ResizeNearestNeighbor, RFFT, IRFFT, Angle, StatelessRandomGetAlg, EnsureShape, CheckNumerics, Roll. #60, #61, #62, #64, #67, #68, #74
  • Support experimental XlaCallModule used by jax2tf. #70, #71
  • Add heuristic for inferring cumulative reductions and avoid potentially expensive reduce_window calls. #55
  • Fix an edge case in StrideSlice. #80
  • Dropped support for python 3.7 (to match JAX). #73

Full Changelog: v0.3.1...v0.3.2