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
0 / 8 versions selected
Downloads (Last 6 months): 0
About
Summary
JMP is a Mixed Precision library for JAX.
Last Updated
Dec 23, 2024 at 00:00
License
Apache-2.0
Total Downloads
1
GitHub Repository
https://github.com/deepmind/jmpDocumentation
https://github.com/deepmind/jmp