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.