# tensorplay.stax

Source: https://www.tensorplay.cn/docs/stax.html

# tensorplay.stax

Static-graph optimization and acceleration. stax traces eager execution into
a static graph, applies compiler passes (constant folding, dead code
elimination, operator decomposition), and lowers the result to the available
backends.

This module is TensorPlay-specific and has no upstream torch counterpart.

## Functions

[stax](generated/tensorplay.stax.stax.html#tensorplay.stax.stax)

Compile one canonical graph and return an executable callable.

[is_available](generated/tensorplay.stax.is_available.html#tensorplay.stax.is_available)

Return whether the native Stax extension is loaded.