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:
11,571
Downloads last week:
64,784
Downloads last month:
352,013