API reference

Generated from package docstrings with mkdocstrings.

Module	Contents
Package root	Public re-exports (anytensor / at)
Core ops	Ordinary Array-API ops, promote, specials
Segment ops	Segment reductions, partition_*, and cache
Semantics	empty_segment_identity
Namespace	TF Array-API shim (array_namespace)
Optional imports	module_if_loaded() — already-imported extras, with callbacks

Graphs and nests have their own sections: Tree, Jraph, Hetero (API pages tree/api, jraph/api, hetero/api). ONNX is the recommended deploy format (ORT); helpers live in the opt-in anytensor.export subpackage, not the core API: ONNX (helpers).

Backends (anytensor.backends) are internal; use public helpers instead of get_backend unless you are extending the library.