Metadata-Version: 2.1
Name: scalax
Version: 0.2.4
Summary: Scaling utilities for JAX.
Home-page: https://github.com/young-geng/scalax
License: Apache-2.0
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3 :: Only
Requires-Python: >=3.8
License-File: LICENSE
Requires-Dist: numpy
Requires-Dist: jax
Requires-Dist: einops
