Closed jiasenlu closed 3 years ago
I am new to jax and thanks for this great codebase. I wonder is there a way to do print debugging inside the model? Or do you have any preferred way? Thanks!
https://jax.readthedocs.io/en/latest/jax.experimental.host_callback.html#jax.experimental.host_callback.id_print
I am new to jax and thanks for this great codebase. I wonder is there a way to do print debugging inside the model? Or do you have any preferred way? Thanks!