update installation setup

Author: chaoming0625

setup.py

@@ -38,6 +38,7 @@
   README = f.read()
 
 # require users to install jaxlib before installing brainpy on Windows platform
+requirements = ['numpy>=1.15', 'jax>=0.3.0', 'tqdm']
 if sys.platform.startswith('win32') or sys.platform.startswith('cygwin'):
   try:
     import jaxlib
@@ -52,8 +53,10 @@
 ----------------------------------------------------------------------
 
 ''') from None
+else:
+  requirements.append('jaxlib>=0.3.0')
 
-
+# installation packages
 packages = find_packages()
 if 'docs' in packages:
   packages.remove('docs')
@@ -71,25 +74,20 @@
   author_email='[email protected]',
   packages=packages,
   python_requires='>=3.7',
-  install_requires=[
-    'numpy>=1.15',
-    'jax>=0.3.0',
-    'jaxlib>=0.3.0',
-    'tqdm',
-  ],
+  install_requires=requirements,
   url='https://github.com/PKU-NIP-Lab/BrainPy',
   project_urls={
     "Bug Tracker": "https://github.com/PKU-NIP-Lab/BrainPy/issues",
     "Documentation": "https://brainpy.readthedocs.io/",
     "Source Code": "https://github.com/PKU-NIP-Lab/BrainPy",
   },
-  keywords='computational neuroscience, '
-           'brain-inspired computation, '
-           'dynamical systems, '
-           'differential equations, '
-           'brain modeling, '
-           'brain dynamics modeling, '
-           'brain dynamics programming',
+  keywords=('computational neuroscience, '
+            'brain-inspired computation, '
+            'dynamical systems, '
+            'differential equations, '
+            'brain modeling, '
+            'brain dynamics modeling, '
+            'brain dynamics programming'),
   classifiers=[
     'Natural Language :: English',
     'Operating System :: OS Independent',