JaxOperatorsMixin¶

class probly.representation.jax_operators.JaxOperatorsMixin[source]¶

Bases: object

Provide array operators through __jax_function__ implementations.

Operators use the corresponding jax_* wrapper as their dispatch key. Implementations control permissions and result types. Augmented assignment uses Python’s ordinary out-of-place operator fallback, matching JAX immutability. Subclasses can override comparison methods; dataclasses must use eq=False to inherit the mixin’s equality operator instead of generating their own.