google-deepmind / jmp

JMP is a Mixed Precision library for JAX.
193Updated last month

Alternatives and similar repositories for jmp:

Users that are interested in jmp are comparing it to the libraries listed below