ColumnStack.jaxify()#

ColumnStack.jaxify()#

Enables JAX NumPy for the node.

Return type

Node