Functions 1
compile
functionFull reference ↗- tensorplay._stax.api.compile(model: Callable[[...], Any] | None = None, *, fullgraph: bool = False, dynamic: bool | None = None, backend: str | Callable[[...], CompiledFn] | None = None, mode: str | None = None, options: dict[str, Any] | None = None, name: str | None = None, disable: bool = False, recompile_limit: int | None = None, isolate_recompiles: bool = False, strict_native: bool = False, dynamic_shapes: Any = None) Callable[[...], Any][source]
Compile a callable through the TensorPlay compiler frontend.
backendmay be a registered name or a callable with the contractbackend(graph_module, example_inputs, **kwargs) -> callable. The frontend caches specializations by input metadata; capture and backend failures are surfaced as compiler errors.
