fridom.framework.utils.jax_utils.inspect_jitted_function# fridom.framework.utils.jax_utils.inspect_jitted_function(func: callable, args: tuple)[source]# Inspect if a jit compiled function has communication operations. Parameters# funccallableThe function to inspect. argstupleThe arguments to pass to the function. Must be a tuple.
