Add progress bars

[astanziola]

There's now support for progress bar in jax scan and while loops!
See https://github.com/jeremiecoullon/jax-tqdm

It should be possible to add progress bars to simulations, which will greatly improve the experience for large problems.

The progress bars should be disabled by default, as they are probably inconvenient when jwave is used for machine learning applications on relatively small problems