Skip to content

jax

Source code in src/einmesh/jax.py 
14
15
def einmesh(pattern: str, *unamed_spaces: SpaceType, **named_spaces: SpaceType) -> jax.Array | tuple[jax.Array, ...]:
    return _einmesh(pattern, *unamed_spaces, backend=JaxBackend(), **named_spaces)