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.
Functions
Compile one canonical graph and return an executable callable. |
|
Return whether the native Stax extension is loaded. |
