Metadata-Version: 2.4
Name: scalax
Version: 0.2.6
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
Dynamic: classifier
Dynamic: home-page
Dynamic: license
Dynamic: license-file
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary
