JMP (JAX Mixed Precision)
JSON →JMP is a Python library from DeepMind that provides abstractions for mixed precision training in JAX. It enables the use of full and half-precision floating-point numbers during model training to reduce memory bandwidth and improve computational efficiency. It is currently at version 0.0.4 and sees active development with new releases addressing JAX compatibility and feature enhancements.
Traffic · last 30 days stale · no recent hits
total hits 9
actors 3 distinct systems
last hit 20d ago AhrefsBot
top countries 🇨🇦 Canada · 🇺🇸 United States · 🇩🇪 Germany
API endpoints
full doc /v1/registry/jmp
install /v1/registry/jmp/install
compatibility /v1/registry/jmp/compatibility