PyPI page
Home page
Author:
DeepMind
License:
Apache 2.0
Summary:
JMP is a Mixed Precision library for JAX.
Latest version:
0.0.4
Required dependencies:
dataclasses
|
numpy
Optional dependencies:
jax
|
jaxlib
Downloads last day:
6,376
Downloads last week:
76,799
Downloads last month:
760,289