Some of the current implementations are either strict Python subsets or very nearly so, but, yes, DSLs are distinct languages, that's what the L stands for.
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.