jmp
JMP is a Mixed Precision library for JAX.
JMP is a Mixed Precision library for JAX.
To install this package, run one of the following:
Mixed precision training [0] is a technique that mixes the use of full and half precision floating point numbers during training to reduce the memory bandwidth requirements and improve the computational efficiency of a given model.
Summary
JMP is a Mixed Precision library for JAX.
Last Updated
Jan 30, 2023 at 16:29
License
Apache-2.0
Total Downloads
27.3K
Supported Platforms
GitHub Repository
https://github.com/deepmind/jmpDocumentation
https://github.com/deepmind/jmp