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:
21,550
Downloads last week:
125,543
Downloads last month:
328,510