stringjax_tools.jit.is_static

Contents

stringjax_tools.jit.is_static#

is_static(arg)#

Heuristic test for whether arg should be treated as JAX-static.

Python scalars, strings, booleans and ordinary configuration objects are treated as static. JAX arrays and NumPy arrays are treated as dynamic.

Return type:

bool