jmp
JMP is a Mixed Precision library for JAX.
Installation
To install this package, run one of the following:
Installation commands are not available for this package.
Usage Tracking
latest
1 / 8 versions selected
Downloads (Last 6 months): 0
About
Summary
JMP is a Mixed Precision library for JAX.
Last Updated
Apr 27, 2026 at 14:42
License
Apache-2.0
GitHub Repository
https://github.com/deepmind/jmpDocumentation
https://github.com/deepmind/jmp