_            _    _        _         _
      /\ \         /\ \ /\ \     /\_\      / /\
      \_\ \       /  \ \\ \ \   / / /     / /  \
      /\__ \     / /\ \ \\ \ \_/ / /     / / /\ \__
     / /_ \ \   / / /\ \ \\ \___/ /     / / /\ \___\
    / / /\ \ \ / / /  \ \_\\ \ \_/      \ \ \ \/___/
   / / /  \/_// / /   / / / \ \ \        \ \ \
  / / /      / / /   / / /   \ \ \   _    \ \ \
 / / /      / / /___/ / /     \ \ \ /_/\__/ / /
/_/ /      / / /____\/ /       \ \_\\ \/___/ /
\_\/       \/_________/         \/_/ \_____\/
python-jmp 0.0.4
Propagated dependencies: python-absl-py@1.4.0 python-dataclasses@0.8 python-jax@0.4.20 python-jaxlib@0.4.20 python-numpy@1.24.4
Channel: guix-science
Location: guix-science/packages/python.scm (guix-science packages python)
Home page: https://github.com/google-deepmind/jmp
Licenses: ASL 2.0
Synopsis: JMP is a mixed precision library for JAX
Description:

This library implements support for mixed precision training in JAX. It provides two key abstractions. These abstractions are mixed precision policies and loss scaling.

Total results: 1