Skip to content

jax

Source code in src/einmesh/jax.py 
14
15
def einmesh(pattern: str, **kwargs: SpaceType) -> jax.Array:
    return _einmesh(pattern, backend=JaxBackend(), **kwargs)  # pyright: ignore[reportReturnType]