Merge branch 'master' into yge/proximal-qr

Author: YigitElma

requirements.txt

@@ -4,12 +4,12 @@ diffrax >= 0.6.0, <= 0.7.2
 equinox >=0.11.10, <=0.13.8
 h5py >= 3.0.0, <= 3.16.0
 interpax >= 0.3.3, < 0.4
-interpax_fft >= 0.0.6, <= 0.0.9
+interpax_fft >= 0.0.6, <= 0.1.0
 jax-finufft >= 1.1.0, <= 1.3.1
 matplotlib >= 3.7.3, <= 3.11.0
 mpmath >= 1.0.0, <= 1.4.1
 netcdf4 >= 1.5.4, !=1.7.4, <= 1.7.5
-numpy >= 1.20.0, <= 2.5
+numpy >= 1.20.0, <= 2.5.1
 nvidia-ml-py >=12.535.77
 optax < 0.3
 orthax < 0.3