eqiora.jax

← Python API index

JAX typed-FFI projection of Eqiora differentiable programs.

Module authority: bindings/python/python/eqiora/jax.py

Shipped stub: bindings/python/python/eqiora/jax.pyi

This module publishes 2 literal __all__ spellings. Member entries below are exact signatures under documented owning types; they do not imply member-level behavioral prose.

JaxProgram

Process-local JAX view of one immutable Eqiora program.

Authority: bindings/python/python/eqiora/jax.py::JaxProgram

python

class JaxProgram:
    def __init__(self, program: DifferentiableProgram) -> None: ...
    @property
    def program(self) -> DifferentiableProgram: ...
    @property
    def input_shape(self) -> tuple[int]: ...
    @property
    def output_shape(self) -> tuple[int]: ...
    def __call__(self, parameters: jax.Array) -> jax.Array: ...

bind

Bind a framework-neutral program to the typed JAX/XLA FFI.

Authority: bindings/python/python/eqiora/jax.py::bind

python

def bind(program: DifferentiableProgram) -> JaxProgram: ...