GitHub repository: vishalbelsare/jmp
Primary language: Python
Description: JMP is a Mixed Precision library for JAX.