traveler.demo.work_location.multithread_op.debug#

multithread_op.debug(choosers, *args, **kwargs)[source]#

Run the vectorized operation without JIT compilation or chunking.

Parameters:

  • choosers (Any) – Chooser data passed as the operation’s first argument.

  • *args (Any) – Additional positional arguments for the wrapped operation.

  • **kwargs (Any) – Additional keyword arguments for the wrapped operation.

Returns:

Result returned by the vectorized debugging operation.

Return type:

jax.Array or dict[str, jax.Array]