Last released Jul 11, 2025
A library for unit scaling in PyTorch, based on the paper 'u-muP: The Unit-Scaled Maximal Update Parametrization.'
Last released Jul 17, 2024
JAX Scalify: end-to-end scaled arithmetic.
Supported by