tf2jax 0.3.2
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