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 ↑233% vs prev 7d
total hits 17
actors 6 distinct systems
last hit 1d ago SERankingBot
top countries 🇩🇪 Germany · 🇺🇸 United States · 🇨🇦 Canada · 🇫🇷 France
API endpoints
full doc /v1/registry/jmp
install /v1/registry/jmp/install
compatibility /v1/registry/jmp/compatibility