How to print information in c++ code?

[south-ocean]

Hi, I knew we can through External callbacks to print information for debug, But i can't print information in C++ code with jax.jit , in fact i have try the print\fprint and soon, it can;t bring information for me. Do anyone have answers?

[jakevdp]

That should work if you're compiling and executing that code – to do so you'll have to build jaxlib from source after making those modifications, and then ensure that you're importing the wheels you built in your Python runtime.

[south-ocean]

yeah，I compile the source code when making those modifications and reinstall it, but it also have no information. Do you have any documentation about this. Can some informations can help me to print in C++ code except for python print.

[jakevdp]

The way you're doing it looks correct to me.