Python support

[greinerth]

First of all, thank you for providing pdebench. ;)

Do you think it is possible to provide Python 3.11 and Python 3.12 support? As far as I know, PyTorch and JAX support the newest interpreters.

[kmario23]

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

Py3.12 support was added in PyTorch 2.2.0

Internally, we have been using the codebase extensively with Py3.10 along with PyTorch v2.2.2.

Although we haven't tested the requested versions, they should work without issues. Have you tested it and encountered problems?

[kmario23]

As for the JAX support, both Py3.11 and Py3.12 versions are supported by jaxlib from 0.4.17 onwards. Here, as well, we have only tested Py3.10 on jaxlib=0.4.23 so far.

Having said that, we can already test it for 3.12 since it has already been a year since this Python release.

[greinerth]

I did not test it yet, but I will tinker with it in the next few days.

[bensworth]

Any update here? I can't use pip install even in python 3.11.3,

$ python
Python 3.11.13 (main, Aug 21 2025, 11:45:17) [GCC 8.5.0 20210514 (Red Hat 8.5.0-28)] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> quit()
$ pip install PDEBench
ERROR: Ignored the following versions that require a different python version: 0.1.0 Requires-Python >=3.9,<3.11
ERROR: Could not find a version that satisfies the requirement PDEBench (from versions: none)

[notice] A new release of pip is available: 25.2 -> 25.3
[notice] To update, run: pip install --upgrade pip
ERROR: No matching distribution found for PDEBench