API symbolsstaxCopyCopy as MarkdownCopy page linkView MarkdownDownload .mdOpenAIOpen in ChatGPTClaudeOpen in ClaudePerplexityOpen in PerplexityGrokOpen in Grok tensorplay.stax.is_available tensorplay.stax.is_available() → bool[source] Return whether the native Stax extension is loaded. Help improve this pageFound an error, an unclear step, or a missing example?Was this page helpful?YesNoSuggest an editReport an issuetensorplay.split_with_sizesPrevious Pagetensorplay.stax.staxCompile one canonical graph and return an executable callable.