Tf2jax

What's Changed
* Returns a callable AnnotatedFunction instead of a plain function. https://github.com/deepmind/tf2jax/pull/35
* Add missing dtype mapping for tf.bfloat16. https://github.com/deepmind/tf2jax/pull/27
* Improve handling of captured arguments for StatelessWhile. https://github.com/deepmind/tf2jax/pull/31
* Minor fix to tf_hub documentation. https://github.com/deepmind/tf2jax/pull/34
* Allow PreventGradient to log warning instead of raise errors. https://github.com/deepmind/tf2jax/pull/36
* Fix handling tf.Module with non-unique variable names. https://github.com/deepmind/tf2jax/pull/40
* Support VariableV2 as placeholder. https://github.com/deepmind/tf2jax/pull/30
* Support FusedBatchNorm. https://github.com/deepmind/tf2jax/pull/32
* Support Rank. https://github.com/deepmind/tf2jax/pull/39


**Full Changelog**: https://github.com/deepmind/tf2jax/compare/v0.2.1...v0.3.0

What's Changed
* Fix broadcasting in BiasAdd. by copybara-service in https://github.com/deepmind/tf2jax/pull/24


**Full Changelog**: https://github.com/deepmind/tf2jax/compare/v0.2.0...v0.2.1

What's Changed
* Changed required tensorflow version from nightly to 2.8.0. by copybara-service in https://github.com/deepmind/tf2jax/pull/8
* Enable custom_gradient support by default. by copybara-service in https://github.com/deepmind/tf2jax/pull/9
* Support NCHW in convolution, pooling and BiasAdd operations. by copybara-service in https://github.com/deepmind/tf2jax/pull/13 and https://github.com/deepmind/tf2jax/pull/14
* Fix a bug with shrink axis in StridedSlice. by copybara-service in https://github.com/deepmind/tf2jax/pull/17
* Support bool in MatrixDiagV3. by copybara-service in https://github.com/deepmind/tf2jax/pull/16
* Support TopKV2. by copybara-service in https://github.com/deepmind/tf2jax/pull/15
* Support XlaRngBitGenerator. copybara-service in https://github.com/deepmind/tf2jax/pull/19


**Full Changelog**: https://github.com/deepmind/tf2jax/compare/v0.1.1...v0.2.0

First release.