jmp - JMP 0.0.4

Changelog:

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

- Python
Published by tomhennigan over 3 years ago

jmp - JMP 0.0.3

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 7b3ae548154625a56a6dbdb03bc07d4f599c37b2.

- Python
Published by tomhennigan over 3 years ago

jmp - JMP 0.0.2

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.

- Python
Published by tomhennigan about 5 years ago