Skip to main content
Pre-release

This release is a pre-release and may not be stable for production use.

Pyrefly einops shape stubs

This package is a PEP 561 stub-only distribution for einops. It provides shape-aware annotations for rearrange, reduce, repeat, and einsum using Pyrefly's type-level shape DSL. The einops.layers.torch Rearrange and Reduce modules preserve the constructor pattern and axis lengths across calls.

The precise annotations currently target PyTorch. The runtime corpus also exercises NumPy and JAX to validate that the shared einops pattern semantics agree across backends.

TODO: Once Pyrefly supports a MapShape type operator, make the annotations generic over array libraries while preserving the input's nominal array type.

Literal axes_lengths keyword arguments are carried into the shape rules, so named input splits and repeated output axes remain tracked. Dynamically unpacked mappings preserve the known rank but use gradual dimensions for unknown axis lengths.

Run the static tests with:

python3 tensor-shapes/pyrefly-einops-stubs/run_pyrefly.py --buck

Metadata

Release files for pyrefly-einops-stubs 1.4.0.dev3

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for pyrefly-einops-stubs 1.4.0.dev3
File Size Uploaded
pyrefly_einops_stubs-1.4.0.dev3.tar.gz 3.3 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for pyrefly-einops-stubs 1.4.0.dev3
File Interpreter ABI Platform
pyrefly_einops_stubs-1.4.0.dev3-py3-none-any.whl Python 3 none any Details

Total release size: 9.1 kB

Release files / pyrefly_einops_stubs-1.4.0.dev3.tar.gz

Download URL pyrefly_einops_stubs-1.4.0.dev3.tar.gz
Size 3.3 kB
Tags Source
SHA-256 checksum
How to use checksums
098b1dc07362ea7da51f06a783e9afc4c7e6c41a1b63d0a1a543eb3126e0fc19
BLAKE2b-256 checksum
How to use checksums
a1f8163205ebd47685b607224e8c80c7c5208492a125e35ba327e9e150c90cb5
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/7.0.0 CPython/3.13.14

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Oct 2, 2026.

Transparency log

Release files / pyrefly_einops_stubs-1.4.0.dev3-py3-none-any.whl

Download URL pyrefly_einops_stubs-1.4.0.dev3-py3-none-any.whl
Size 5.7 kB
Tags Python 3
SHA-256 checksum
How to use checksums
30f50c432c9e7afd0318ee7e1617ea68d777beab484b88f434ae8894794ef281
BLAKE2b-256 checksum
How to use checksums
ec42b984db29c0a21092e772606a1412c2e41ca2e7423ed70ad82531f50ef19f
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/7.0.0 CPython/3.13.14

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Oct 2, 2026.

Transparency log
Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page