jax

f0540d11 - Use `()` instead of `tuple()` and `{}` instead of `dict()`