jax.experimental.pallas.debug_check

Contents

jax.experimental.pallas.debug_check#

jax.experimental.pallas.debug_check(condition, message)[source]#: Check the condition if enable_debug_checks() is set, otherwise do nothing.