Mario
Hey @greinerth, appreciate the PR! Can you please have a look at the recent commits and resolve conflicts?
@greinerth I've had time to look into it. There are issues with relative imports of `vorticity.compute_spectral_vorticity_jnp` in `data_gen/velocity2vorticity.py`. `vorticity.py` should be in the same tree level, which is now in...
Hey @greinerth, I had a few hours today, and I tried merging the PR and fixing some Ruff linter errors, but there's still a lot to fix. We especially need...
Perhaps we should consider using a logger (e.g., python's `logging`) since Ruff complains about the `print()` statements.
Hey @greinerth, Thanks for your interest in the benchmark. Sure, we would be glad to provide support for these Python versions. [Py3.11 has been supported in PyTorch from v2.0.1 onwards](https://anaconda.org/pytorch/pytorch/files?version=2.0.1)...
As for the JAX support, [both `Py3.11` and `Py3.12` versions are supported by `jaxlib` from `0.4.17`](https://storage.googleapis.com/jax-releases/jax_cuda_releases.html) onwards. Here, as well, we have only tested `Py3.10` on `jaxlib=0.4.23` so far. Having...