v0.2.0

What's Changed

• AGC and mixed precision are now compatible
• Support for both float16 and bfloat16 on GPU and TPU, respectively
• Set default accum_steps in GAModelWrapper to 1
• Improved documentation regarding usage of AGC, GPU/TPU, and recommended model format (SavedModel)
• Support for 3D operations in AGC, such as Conv3D, by @dbouget in #17
• Corrected mean reduction to before gradient computation @dbouget in #2

New Contributors

• @dbouget made their first contribution in #17

Full Changelog: v0.1.5...v0.2.0