piwheels

scalax

Scaling utilities for JAX.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install scalax

Dependencies

• einops
• numpy
• jax

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
0.2.6	2025-11-04
scalax-0.2.6-py3-none-any.whl (11 KB)
0.2.5	2024-10-16
scalax-0.2.5-py3-none-any.whl (11 KB)
0.2.4	2024-08-13
scalax-0.2.4-py3-none-any.whl (11 KB)
0.2.3	2024-06-26
scalax-0.2.3-py3-none-any.whl (11 KB)
0.2.2 yanked	2024-06-22
scalax-0.2.2-py3-none-any.whl (11 KB)
0.2.1	2024-03-07
scalax-0.2.1-py3-none-any.whl (11 KB)
0.2.0	2024-02-29
scalax-0.2.0-py3-none-any.whl (13 KB)
0.1.5	2024-02-28
scalax-0.1.5-py3-none-any.whl (13 KB)
0.1.3	2024-02-06
scalax-0.1.3-py3-none-any.whl (12 KB)
0.1.1	2024-02-03
scalax-0.1.1-py3-none-any.whl (11 KB)
0.1.0	2023-09-19
scalax-0.1.0-py3-none-any.whl (11 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue