About Anaconda Help Download Anaconda

JMP is a Mixed Precision library for JAX.

copied from cf-staging / jmp

Installers

  • noarch v0.0.4

conda install

To install this package run one of the following:
conda install conda-forge::jmp

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.


© 2024 Anaconda, Inc. All Rights Reserved. (v4.0.6) Legal | Privacy Policy