SPI-2404: Update Python packages

Update Python packages requested by Luca:

jax 0.4.8->0.4.14

jaxlib 0.4.7->0.4.14

ml_dtypes 0.1.0->0.2.0

pytorch_lightning 2.0.2->2.0.6

tensorflow 2.12.0->2.13.0

transformers 4.30.2->4.31.0

xgboost 1.7.3->1.7.6

Mention SPI-2404