tf2jax 0.3.3

shaobohou released this 10 Feb 20:21

· 112 commits to main since this release

6bba873

What's Changed

Make TF2JAX run on M1 macs. #81
Use input_signature when available. #83
Preemptive fix for upcoming JAX changes. #85, #86
Preemptive fix for polymorphic shape support. #88
Re-enable native lowering test for nested custom gradient. #91
Support XlaReducePrecision. #93

Full Changelog: v0.3.2...v0.3.3

Assets 2