Skip to content

Releases: google-deepmind/jmp

JMP 0.0.4

30 Jan 12:46
Compare
Choose a tag to compare

Changelog:

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

JMP 0.0.3

20 Jan 13:11
Compare
Choose a tag to compare

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.

JMP 0.0.2

15 Apr 09:38
Compare
Choose a tag to compare

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.