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:
4,948
Downloads last week:
39,208
Downloads last month:
237,203