jax_operators

Python operators expressed through probly’s JAX function override protocol.

Classes

JaxOperatorsMixin

Provide array operators through __jax_function__ implementations.