I'm alway shocked when people in this line of work either take things at face-value or just lie by omission/imprecision. this is not python breakpoints - it's a wholeass other system they had to re-roll that emulates python breakpoints:

> Unlike pdb, you will not be able to step through the execution, but you are allowed to resume it.

you think it works with pycharm/vscode/any other tooling? spoiler alert: of course not.

so no in JAX you can not.