CMD + K

jmp

Community

JMP is a Mixed Precision library for JAX.

Copied fromcf-staging / jmp

Installation

To install this package, run one of the following:

Conda
$conda install conda-forge::jmp

Usage Tracking

0.0.4
0.0.3
0.0.2
3 / 8 versions selected
Downloads (Last 6 months): 0

Description

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.

About

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

noarch