requirements updated

requirements.txt

-absl-py>=0.13.0
-astroid>=2.7.3
-backcall>=0.2.0
-Box2D>=2.3.10
-certifi>=2021.5.30
-cloudpickle>=1.6.0
-cycler>=0.10.0
-decorator>=5.0.9
-flatbuffers>=2.0
-future>=0.18.2
-gym>=0.19.0
-imageio>=2.9.0
-ipython>=7.27.0
-ipython-genutils>=0.2.0
-isort>=5.9.3
+-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+jaxlib==0.1.73+cuda11.cudnn805
 jax==0.2.20
-jaxlib==0.1.71
-joblib>=1.0.1
-kiwisolver>=1.3.2
-lazy-object-proxy>=1.6.0
-matplotlib>=3.4.3
-matplotlib-inline>=0.1.3
-mccabe>=0.6.1
-networkx>=2.6.2
-numpy>=1.21.2
-opt-einsum>=3.3.0
-parso>=0.8.2
-pexpect>=4.8.0
-pickleshare>=0.7.5
-Pillow>=8.3.2
-platformdirs>=2.3.0
-prompt-toolkit>=3.0.20
-ptyprocess>=0.7.0
-pygame>=2.0.1
-pyglet>=1.5.20
-Pygments>=2.10.0
-pylint>=2.10.2
-pyqtgraph==0.12.2
-pyqt5==5.12
-python-dateutil>=2.8.2
-PyWavelets>=1.1.1
-PyYAML>=5.4.1
-scikit-fmm>=2021.7.8
-scikit-learn>=0.24.2
-scipy>=1.7.1
-shiboken2>=5.15.2
-six>=1.16.0
-threadpoolctl>=2.2.0
-tifffile>=2021.8.30
-toml>=0.10.2
-traitlets>=5.1.0
-wcwidth>=0.2.5
-wrapt>=1.12.1
+numpy
+scikit-fmm
+scikit-learn
+scipy

setup.cfg

@@ -10,16 +10,5 @@ zip_safe = False
 include_package_data = True
 packages = find:
 install_requires =
-  numpy >= 1.21.2
-  scipy >= 1.7.1
-  jax == 0.2.20
-  jaxlib == 0.1.71
-  PyYAML >= 5.4.1
-  scikit-fmm >= 2021.7.8
-  opencv-python >= 4.6.0.66
-python_requires = >=3.8
-scripts =
-    demo.py
-
 [options.package_data]
 * = config/*

setup.py

@@ -11,14 +11,4 @@
 
 import setuptools
 
-setuptools.setup(
-    install_requires=[
-        'numpy >= 1.21.2',
-        'scipy >= 1.7.1',
-        'jax == 0.2.20',
-        'jaxlib == 0.1.71',
-        'PyYAML >= 5.4.1',
-        'scikit-fmm >= 2021.7.8',
-        'opencv-python >= 4.6.0.66',
-        ]
-)
+setuptools.setup()