function jit

jit(fn: PyFunc | None = None, *, signature: Signature | None = None, cache: bool = False) -> JitFunction | functools.partial[JitFunction]

Source: packages/arxjit/src/arxjit/core.py:75

Compile a pure Python function through the Arx stack.

Usable bare as @jit or with options as @jit(signature=i64(i64, i64), cache=True). Records the requested signature and returns a callable wrapper. When called with options it returns a partial that wraps the function next.

Parameters

• fn — Type: PyFunc | None; The function to wrap, when used as a bare decorator.

• signature — Type: Signature | None; The explicit call signature.

• cache — Type: bool; Whether to cache compiled artifacts.

Returns

Type: JitFunction | functools.partial[JitFunction]