Reference guides
Copy
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
Compile one canonical graph and return an executable callable. |
|
Return whether the native Stax extension is loaded. |
Help improve this page
Found an error, an unclear step, or a missing example?
Was this page helpful?