Changelog:

• Avoid triggering warning from use of jax.tree_leaves.
• Avoids triggering warnings when using DynamicLossScale.

Changelog:

• Add support for Python 3.11 #30.
• Drop support for Python 3.7.
• DynamicLossScale supports configurable minimum value #18.
• DynamicLossScale now warns if a non-floating type is used #25.
• cast_to_* now support NumPy array (as well as JAX Array) as input 7b3ae54.

Initial release of JMP.

Changelog:

• Add jmp.Policy abstraction and jmp.get_policy(..) factory.
• Add jmp.LossScale and three implementations thereof (noop, static and dynamic).
• Add various utilities (jmp.all_finite) to support common tasks in mixed precision codebases.