Yet the documentation for jax.debug.print clearly states: fmt (str) – A format string, e.g. "hello {x}", that will be used to format input arguments, like str.format. See the Python docs on string ...