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:
29,673
Downloads last week:
93,461
Downloads last month:
465,328