quax-blocks

Building blocks for `Quax` classes

quax enables JAX to work with custom array-ish objects. This library provides the building blocks, like comparison operators, for building quax-compatible classes.

Installation

pip install quax-blocks

Documentation

Rich Comparison Operators

Comparison Operator	Lax Operator	NumPy Operator
All Comparisons	`LaxComparisonMixin`	`NumpyComparisonMixin`
`__eq__`	`LaxEqMixin`	`NumpyEqMixin`
`__ne__`	`LaxNeMixin`	`NumpyNeMixin`
`__lt__`	`LaxLtMixin`	`NumpyLtMixin`
`__le__`	`LaxLeMixin`	`NumpyLeMixin`
`__gt__`	`LaxGtMixin`	`NumpyGtMixin`
`__ge__`	`LaxGeMixin`	`NumpyGeMixin`

Binary Operators

Binary Operator	Lax Operator	NumPy Operator
All Binary Operations	`LaxBinaryOpsMixin`	`NumpyBinaryOpsMixin`
All Float Operations	`LaxMathMixin`	`NumpyMathMixin`
`__add__`	`LaxAddMixin`	`NumpyAddMixin`
`__radd__`	`LaxRAddMixin`	`NumpyRAddMixin`
`__sub__`	`LaxSubMixin`	`NumpySubMixin`
`__rsub__`	`LaxRSubMixin`	`NumpyRSubMixin`
`__mul__`	`LaxMulMixin`	`NumpyMulMixin`
`__rmul__`	`LaxRMulMixin`	`NumpyRMulMixin`
`__matmul__`	`LaxMatMulMixin`	`NumpyMatMulMixin`
`__rmatmul__`	`LaxRMatMulMixin`	`NumpyRMatMulMixin`
`__truediv__`	`LaxTrueDivMixin`	`NumpyTrueDivMixin`
`__rtruediv__`	`LaxRTrueDivMixin`	`NumpyRTrueDivMixin`
`__floordiv__`	`LaxFloorDivMixin`	`NumpyFloorDivMixin`
`__rfloordiv__`	`LaxRFloorDivMixin`	`NumpyRFloorDivMixin`
`__mod__`	`LaxModMixin`	`NumpyModMixin`
`__rmod__`	`LaxRModMixin`	`NumpyRModMixin`
`__divmod__`	`LaxDivModMixin`	`NumpyDivModMixin`
`__rdivmod__`	`LaxRDivModMixin`	`NumpyRDivModMixin`
`__pow__`	`LaxPowMixin`	`NumpyPowMixin`
`__rpow__`	`LaxRPowMixin`	`NumpyRPowMixin`
All Bitwise Operations	`LaxBitwiseMixin`	`NumpyBitwiseMixin`
`__lshift__`	`LaxLShiftMixin`	`NumpyLShiftMixin`
`__rlshift__`	`LaxRLShiftMixin`	`NumpyRLShiftMixin`
`__rshift__`	`LaxRShiftMixin`	`NumpyRShiftMixin`
`__rrshift__`	`LaxRRShiftMixin`	`NumpyRRShiftMixin`
`__and__`	`LaxAndMixin`	`NumpyAndMixin`
`__rand__`	`LaxRAndMixin`	`NumpyRAndMixin`
`__xor__`	`LaxXorMixin`	`NumpyXorMixin`
`__rxor__`	`LaxRXorMixin`	`NumpyRXorMixin`
`__or__`	`LaxOrMixin`	`NumpyOrMixin`
`__ror__`	`LaxROrMixin`	`NumpyROrMixin`

Unary Operators

Unary Operator	Lax Operator	NumPy Operator
All Unary Operations	`LaxUnaryMixin`	`NumpyUnaryMixin`
`__pos__`	`LaxPosMixin`	`NumpyPosMixin`
`__neg__`	`LaxNegMixin`	`NumpyNegMixin`
`__abs__`	`LaxAbsMixin`	`NumpyAbsMixin`
`__invert__`	`LaxInvertMixin`	`NumpyInvertMixin`

Rounding Operators

Unary Operator	Lax Operator	NumPy Operator
`__round__`	`LaxRoundMixin`	`NumpyRoundMixin`
`__trunc__`	`LaxTruncMixin`	`NumpyTruncMixin`
`__floor__`	`LaxFloorMixin`	`NumpyFloorMixin`
`__ceil__`	`LaxCeilMixin`	`NumpyCeilMixin`

Containers

Container Operator	Lax Operator	NumPy Operator
`__len__`	`LaxLenMixin`	`NumpyLenMixin`
`__getitem__`	Not Implemented	`NumpyGetItemMixin`
`__length_hint__`	`LaxLengthHintMixin`	`NumpyLengthHintMixin`

Copy Operators

Copy Operator	NumPy Operator
`__copy__`	`NumpyCopyMixin`
`__deepcopy__`	`NumpyDeepCopyMixin`

Development

We welcome contributions!

Citation

If you found this library to be useful and want to support the development and maintenance of lower-level utility libraries for the scientific community, consider citing this work.

Metadata

Release files for quax-blocks 0.5.0

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

Source distribution (sdist)

Source distribution for quax-blocks 0.5.0
File	Size	Uploaded
quax_blocks-0.5.0.tar.gz	138.1 kB	Jul 31, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for quax-blocks 0.5.0
File	Interpreter	ABI	Platform	Reset
quax_blocks-0.5.0-py3-none-any.whl 24.8 kB Jul 31, 2026	Python 3	none	any	Details

Total release size: 162.9 kB

Release files / quax_blocks-0.5.0.tar.gz

Download URL	quax_blocks-0.5.0.tar.gz
Size	138.1 kB
Tags	Source
SHA-256 checksum How to use checksums	`874effa5777aaf48a8bc1a2ef5397d93131ac3f00166b69da5d1d5f7e71232ae`
BLAKE2b-256 checksum How to use checksums	`4540a9197e785f32ab31037266f4834d44d36f1ea8676f52f1af00a719c802df`
Upload date	Jul 31, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	Yes
Uploaded via	`twine/6.1.0 CPython/3.13.13`

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.

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/GalacticDynamics/quax-blocks
Publishing commit: github.com/GalacticDynamics/quax-blocks/tree/3952103526a3210f8c7f35adb2e270fb5cbf56bf

Workflow

Publishing configuration: github.com/GalacticDynamics/quax-blocks/blob/3952103526a3210f8c7f35adb2e270fb5cbf56bf/.github/workflows/cd.yml
Publishing logs: github.com/GalacticDynamics/quax-blocks/actions/runs/30671106825/attempts/1

Artifact

Subject: quax_blocks-0.5.0.tar.gz
SHA-256 checksum: 874effa5777aaf48a8bc1a2ef5397d93131ac3f00166b69da5d1d5f7e71232ae Verifying attestations

Release files / quax_blocks-0.5.0-py3-none-any.whl

Download URL	quax_blocks-0.5.0-py3-none-any.whl
Size	24.8 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`03e9deba2d6c6b51be47108ec52e4b991b7f466529aadab4abddf6fa68be2d60`
BLAKE2b-256 checksum How to use checksums	`00e2f8e50cc72cf5e92d7056d70d396dbee01172133bd081d6d85026c56916c8`
Upload date	Jul 31, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	Yes
Uploaded via	`twine/6.1.0 CPython/3.13.13`

Provenance

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/GalacticDynamics/quax-blocks
Publishing commit: github.com/GalacticDynamics/quax-blocks/tree/3952103526a3210f8c7f35adb2e270fb5cbf56bf

Workflow

Publishing configuration: github.com/GalacticDynamics/quax-blocks/blob/3952103526a3210f8c7f35adb2e270fb5cbf56bf/.github/workflows/cd.yml
Publishing logs: github.com/GalacticDynamics/quax-blocks/actions/runs/30671106825/attempts/1

Artifact

Subject: quax_blocks-0.5.0-py3-none-any.whl
SHA-256 checksum: 03e9deba2d6c6b51be47108ec52e4b991b7f466529aadab4abddf6fa68be2d60 Verifying attestations