search

pybamm-team / PyBaMM

Fast and flexible physics-based battery models in Python
https://www.pybamm.org/
BSD 3-Clause "New" or "Revised" License
1.1k stars 543 forks source link

Using jax solver returns error #1641

Closed raghuramshankar closed 3 years ago

raghuramshankar commented 3 years ago

Describe the bug Using Jax solver throws an error.

To Reproduce Minimal reproduce-able example:

import pybamm

"""models and parameters"""
model = pybamm.lithium_ion.SPMe()

"""solver"""
sol = pybamm.JaxSolver(method='RK45', atol=1e-6, rtol=1e-6)

"""simulate cell"""
model.convert_to_format = 'jax'
sim = pybamm.Simulation(model, solver=sol)
sim.solve([0, 360])

Output:

/home/raghurams/repos/optimal-battery-charging/tutorials/jax_issue.py
Traceback (most recent call last):
  File "/home/raghurams/repos/optimal-battery-charging/tutorials/jax_issue.py", line 12, in <module>
    sim.solve([0, 360]) 
  File "/home/raghurams/.local/lib/python3.9/site-packages/pybamm/simulation.py", line 399, in solve
    self._solution = solver.solve(
  File "/home/raghurams/.local/lib/python3.9/site-packages/pybamm/solvers/base_solver.py", line 567, in solve
    self.set_up(model, ext_and_inputs, t_eval)
  File "/home/raghurams/.local/lib/python3.9/site-packages/pybamm/solvers/base_solver.py", line 383, in set_up
    model.y0 = init_eval(inputs)
  File "/home/raghurams/.local/lib/python3.9/site-packages/pybamm/solvers/base_solver.py", line 956, in __call__
    return self._function(0, self.y_dummy, inputs=inputs).flatten()
  File "/home/raghurams/.local/lib/python3.9/site-packages/pybamm/expression_tree/operations/evaluate.py", line 459, in evaluate
    result = self._jit_evaluate(self._constants, t, y, y_dot, inputs, known_evals)
ValueError: Non-hashable static arguments are not supported. An error occured while trying to hash an object of type <class 'list'>, [DeviceArray([[0. ],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.8],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [0.6],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ],
             [1. ]], dtype=float64)]. The error was:
TypeError: unhashable type: 'list'

Expected behaviour Should be able to solve the experiment using jax solver. I read the pybamm documentation and saw that the extra code model.convert_to_format = 'jax' is necessary while using jax solver. Tried searching a lot but not sure if more code is required for the same.

martinjrobins commented 3 years ago

thanks for the bug report @raghuramshankar. Can you please post the version number of the pybamm, jax and jaxlib modules? i.e.

→ pip list | grep -E 'pybamm|jax|jaxlib'
jax                           0.2.12
jaxlib                        0.1.65
pybamm                        0.4.0     /home/mrobins/git/PyBaMM
raghuramshankar commented 3 years ago

Hi,

Thank you for your response. Here are my library versions:

pip list | grep -E 'pybamm|jax|jaxlib' 
jax                 0.2.19
jaxlib              0.1.70+cuda111
pybamm              0.2.4.post3

I am aware that I am not using the latest version of pybamm. I had some issues in installing the latest version of pybamm and I have reported them in https://github.com/pybamm-team/PyBaMM/issues/1608 and https://github.com/pybamm-team/PyBaMM/issues/1470.

I checked the changelog and found that there were no changes or bug fixes for the jax_solver since v0.2.4. However, I wonder if not using the latest version might affect the jax simulations.

martinjrobins commented 3 years ago

I see, the setup.py in pybamm version 0.2.4 listed the jax dependencies as

"jax>=0.1.68", "jaxlib>=0.1.47",

however they should have been pinned to the exact version we support. Try installing "jax=0.1.68" and "jaxlib=0.1.47" in your environment and hopefully this should fix it.

i.e.

pip install -Iv jax==1.1.68
raghuramshankar commented 3 years ago

Hi,

It seems like only jaxlib>=0.1.60 is available to be installed via pip. Here is the error I got:

pip install -Iv jaxlib==0.1.47
Using pip 21.1.3 from /home/raghurams/repos/optimal-battery-charging/venv/lib/python3.9/site-packages/pip (python 3.9)
Non-user install because user site-packages disabled
Created temporary directory: /tmp/pip-ephem-wheel-cache-4y2oaj1z
Created temporary directory: /tmp/pip-req-tracker-ppu80go8
Initialized build tracking at /tmp/pip-req-tracker-ppu80go8
Created build tracker: /tmp/pip-req-tracker-ppu80go8
Entered build tracker: /tmp/pip-req-tracker-ppu80go8
Created temporary directory: /tmp/pip-install-3zjwzvyc
1 location(s) to search for versions of jaxlib:
* https://pypi.org/simple/jaxlib/
Fetching project page and analyzing links: https://pypi.org/simple/jaxlib/
Getting page https://pypi.org/simple/jaxlib/
Found index url https://pypi.org/simple
Looking up "https://pypi.org/simple/jaxlib/" in the cache
Request header has "max_age" as 0, cache bypassed
Starting new HTTPS connection (1): pypi.org:443
https://pypi.org:443 "GET /simple/jaxlib/ HTTP/1.1" 304 0
  Skipping link: none of the wheel's tags (cp36-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/b1/1c/4c17ea5e0962c66f37a26572adde35d053e580cd15a1df3987bfdcf769c0/jaxlib-0.1.60-cp36-none-macosx_10_9_x86_64.whl#sha256=eab6715820de2d3be99bcf1bd6e74fc062288eec921aec6ad061eff215fcbd23 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp36-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/dd/d5/fe4aa57ac61929cfc07da094feda8b30c75a975d5fe7cdbb82ad3f3c7396/jaxlib-0.1.60-cp36-none-manylinux2010_x86_64.whl#sha256=21bbe2b328a728accb6ee6d370ec948317becc0e32c4921ff5240260fe21206a (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/71/75/9709f7b192936f340e1a332e205f55a752508a5971c00be0cde20095816e/jaxlib-0.1.60-cp37-none-macosx_10_9_x86_64.whl#sha256=f74edc451531a7d12c49c18344db09183ee93c7533c58c120d5444f5d5f8da73 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/79/77/57c2eff18f1cb006fab4d25b074fe9ad68f9a54e365a576651d80e4665f8/jaxlib-0.1.60-cp37-none-manylinux2010_x86_64.whl#sha256=231539132e6322a1d65329bbadcb56cbd2495f584a759437881ee4c163250247 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/ae/de/7d67082ad408d592dda52acd9ba90aecbb335ee5345e60da4dcbda386bca/jaxlib-0.1.60-cp38-none-macosx_10_9_x86_64.whl#sha256=4f0accb342d2559b2c199c9d9267ea011c543060213ab9fd03ffb75f6c0150b9 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/06/80/ce3138bfea4d717e5d529dde01d9387978a3227370ee7d16214e1883249c/jaxlib-0.1.60-cp38-none-manylinux2010_x86_64.whl#sha256=7bff48c253c7bf5ad0986af8e14c981bc5f18855f6eef16f08da6ca915b092f4 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp39-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/02/11/5729188b0e731c65dd556514c572756f0e19051eec44f30c48718af9d0d4/jaxlib-0.1.60-cp39-none-macosx_10_9_x86_64.whl#sha256=3ebe62f1a389b59e0c1661a6d2c14762b68c5864bc07d7de26d81a1bd29b2060 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Found link https://files.pythonhosted.org/packages/5d/e2/4f06a3198ecafbc92ef619a120950129910a1b456a5151799516344e5e21/jaxlib-0.1.60-cp39-none-manylinux2010_x86_64.whl#sha256=5afbf0a176f0b39e349fe3942fdd06f97093130961ae057d0e567fa247746088 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6), version: 0.1.60
  Skipping link: none of the wheel's tags (cp36-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/91/a4/1c8ea78495754cf8f1c4c4ed0100486f5bf62f96f778f2fba7cddcb991ce/jaxlib-0.1.61-cp36-none-macosx_10_9_x86_64.whl#sha256=d88866813d6438cbb2f545b6983091525bdbe8633dbbc3ea7842b2e5d561ec6b (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp36-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/1b/49/f3e6e9c6e5833fde2be8775ed87b5712beb3620c7ea66ac917d25fbfbaa0/jaxlib-0.1.61-cp36-none-manylinux2010_x86_64.whl#sha256=0ef0449dad105144fde97aaae33585b7186c6fb80390e77771a745cc9a937357 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/bb/a2/2c2cc5f913fe4a401fa22c2a9f51b30e4af90ba0d69294a504323df1492a/jaxlib-0.1.61-cp37-none-macosx_10_9_x86_64.whl#sha256=abbd90228b48add038969f116ea85dd10a9efd8db185bb1a1a6339e48e0c2a41 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/22/88/0b77a4f185b4b5c568f6f935f8029192166912e58bec22f0932c703fc10f/jaxlib-0.1.61-cp37-none-manylinux2010_x86_64.whl#sha256=cff3a541567b5723314687f71dbcfd349faf78d362b0a6e96a5f2c354015b59c (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/d0/8e/4acd22a5b30a8968e8f4c4345f020d4c011c112347e6f0d28186471c35fa/jaxlib-0.1.61-cp38-none-macosx_10_9_x86_64.whl#sha256=3f53fb31297cf07d43ebf0a434313861f90e6ac685b98996bc1b77b48c48e097 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/84/47/f51a4bf2dfbcf49ba67895e9456923b5d9bb20f1b508e2f8e59b07dbc66e/jaxlib-0.1.61-cp38-none-manylinux2010_x86_64.whl#sha256=e0b81fcf77579702b2068bafe379ba04d153bd19239ffc2fd2cbe8308ffe3953 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp39-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/0d/72/78730f08ca83adc3deda6c39fe078f60459dbc1b85e4de576925a2f12d43/jaxlib-0.1.61-cp39-none-macosx_10_9_x86_64.whl#sha256=72d08423116bfa4fdd58f0264e73feb4263643b0407a0fc619246d2c9f9b1366 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Found link https://files.pythonhosted.org/packages/07/b9/3aaae06dabf2d75486a3bc0e93fa2c7b67b5fd3f6c5e8f1a4ea71b28d23e/jaxlib-0.1.61-cp39-none-manylinux2010_x86_64.whl#sha256=713f2799158128f7af45d8cfb627bd585f142da9314485f345ccd80bd2eaf8c5 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6), version: 0.1.61
  Skipping link: none of the wheel's tags (cp36-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/a5/ad/29956f1d56b33718abc8a0bb6c24ecef149953beeea26b06221c17b358e0/jaxlib-0.1.62-cp36-none-macosx_10_9_x86_64.whl#sha256=9b58be444ef3147913e4117ab34eae80a41a785291380b113ab6a179157de222 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp36-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/71/42/5a97178a0e48d3647567917e72585c44d4d1cb521753af3b979e1c21e821/jaxlib-0.1.62-cp36-none-manylinux2010_x86_64.whl#sha256=8ae71b54b86f140a840b8c097b015e6fde5c323cc7b2ea716b2707a66370b01e (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/d5/97/52bff772178d26439e645126c5e886e792294e038bb3766c6663f6c34d14/jaxlib-0.1.62-cp37-none-macosx_10_9_x86_64.whl#sha256=58cf9a47e346ca63a3bb6d7ce6b15531fa9ae9bd720cdf99223acb397e1ac26f (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/7e/75/30f1c643b7edb1309b6d748809042241737fe43127cb41754266eca79250/jaxlib-0.1.62-cp37-none-manylinux2010_x86_64.whl#sha256=71202f39a02e7286a5371b75fd58f0131514481dc230a0635447f8470359d92c (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/82/62/d4befd9e93bfc54ec6631accc1c802877e7c3022e9e2e4360abacbc1f4bf/jaxlib-0.1.62-cp38-none-macosx_10_9_x86_64.whl#sha256=7e7652ce36a23ca2707cf1818db8761583e64238ca29592f630d620d58f973c4 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/6d/74/d7bcea75ec1af9e60df4ceb29be802e4824a2fb5d7836a9595383d1781a4/jaxlib-0.1.62-cp38-none-manylinux2010_x86_64.whl#sha256=5a615327a4e97b99090bfdf3b24053f9ca8439a3b22d6455450589b7c5e8f95d (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp39-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/e4/38/f17f18e15319ac7419565ab7bf4acc428f59a50be4ba8f9e8bcb6f4358bb/jaxlib-0.1.62-cp39-none-macosx_10_9_x86_64.whl#sha256=42d6aff8a7262da6f69b55ecf64847ec741b164e179168f5be6008c23c956edf (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Found link https://files.pythonhosted.org/packages/d2/bb/c8263c2b0a6d95fa4e093ee59a34b6ff50a7d346898dcb25ad62905ada1d/jaxlib-0.1.62-cp39-none-manylinux2010_x86_64.whl#sha256=bb878ea37bacd121ac183144998a466aa119859930b787393ea74f7034e0660b (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6), version: 0.1.62
  Skipping link: none of the wheel's tags (cp36-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/db/18/79edf09b915737c62ecf4934b0e6f46c6a58d4de6852be187d575690082f/jaxlib-0.1.63-cp36-none-macosx_10_9_x86_64.whl#sha256=66c486e1e3936a016d5631b2e413961cf7c9f22b5ebfacca50b25e982d4fc241 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp36-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/14/36/55237306798ca460b390fd9c551e93c70fbe19f335aaea59bdbf9cf7c78c/jaxlib-0.1.63-cp36-none-manylinux2010_x86_64.whl#sha256=71befb418612d4862591f0ece7b019d20585182a50df51c49d48eda9347006a9 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/7d/24/ad82e0dd08f56d61c708f1f36b51a1df29f19908f5cd3617a29f30af07cf/jaxlib-0.1.63-cp37-none-macosx_10_9_x86_64.whl#sha256=c150b6a241f55d6c132f6a164c35940ab905ebb52e7cc7c739be32405e1aab5d (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/16/a3/bd0e19706c5172f8670cafc7ca1176e6334616c5279e0cb2666eda3ae3ea/jaxlib-0.1.63-cp37-none-manylinux2010_x86_64.whl#sha256=64b113e5692b747a4aab930bce10cd4cf06d2b49ac422a5de7c3b0d830d7ec95 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/62/65/e0466d83b0a5aae04152e2403a2b06b84ab66fdc71b6ab20f1a4ba022fc6/jaxlib-0.1.63-cp38-none-macosx_10_9_x86_64.whl#sha256=e6b956e0aed05f586be619275214e0c401f6a5e11bdda3b2de3973fccab2d9bd (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/fa/5c/321a121dd12a5ad5d4c2d46695483df5493e7e7c66bdb90bc3d48a859479/jaxlib-0.1.63-cp38-none-manylinux2010_x86_64.whl#sha256=692081a31be3eb1329b59029982a7fd6bb94271ece0bdede0e7c19d075f0f40b (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp39-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/4b/4d/bde4acfa5874df8a609fc10c2c365c343847ec9b63fb4d22674bf8244dce/jaxlib-0.1.63-cp39-none-macosx_10_9_x86_64.whl#sha256=b085d30a29e9a8a9669dbcc3886059665055da488e21e2dfeb2e4c49ce16b150 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Found link https://files.pythonhosted.org/packages/0e/c4/41ab3f41f4e065dfdadedbf70c11a5f6802adeb1ea9227db63786f084a29/jaxlib-0.1.63-cp39-none-manylinux2010_x86_64.whl#sha256=b059c6343959e31a3c58ee9970df112b0c9c014a986ba4cb24ed52f5e76fbf7a (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6), version: 0.1.63
  Skipping link: none of the wheel's tags (cp36-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/3e/21/5f5ada834f6a8b9f0240d3848d80318bab01901dfa5aa9a96cd86e6224c6/jaxlib-0.1.64-cp36-none-macosx_10_9_x86_64.whl#sha256=b2a7cd42032c3eefd6b03bbfb026f7f76ac290c520bc6d7e040ae83f0f84e4f6 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp36-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/f5/c5/4e8dbc2bcf99ae955466fbda337bd0ef251521ce1b01a3326acf70e63d29/jaxlib-0.1.64-cp36-none-manylinux2010_x86_64.whl#sha256=4f60212d926d721b44898599dba88290f21a2bfb5512d14df4a7a81bf0ec06c9 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/c0/38/07d9437704fed69b8f2fc69bce663d4018424a7df5a45eb8075dba67da57/jaxlib-0.1.64-cp37-none-macosx_10_9_x86_64.whl#sha256=72099ac6940728b9e9f76cc81bb547454d8bcbeba1a082a2abb6f0dc201ead70 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/5c/09/8d35f8f51f38030e808d885d77f90b955639c7d60e26952c45c25b6a9069/jaxlib-0.1.64-cp37-none-manylinux2010_x86_64.whl#sha256=362f05b82934669f817eb2dff0b5a807ab513dc0b2387cb58280664839118b81 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/d6/60/cc963006a800f9e8d9afb3d10b8c5d6995181cafa06538943a97f51431bb/jaxlib-0.1.64-cp38-none-macosx_10_9_x86_64.whl#sha256=824a784a770e0a2d1f71322f52a560b4da8dad8009950df1438e3b6f070eb343 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/a0/e6/90f39d8afbac6ca1e018021f02d81b3fcd6a7b08563dc5d7b637cb51a402/jaxlib-0.1.64-cp38-none-manylinux2010_x86_64.whl#sha256=6993ac9b0a985f473b75e142651dafad7186a98dd39e5ad3663addb74c6b9317 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp39-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/e5/09/0b7dde15cbfca1d119a79e7b7aef0ef6334c027b35a6acd055c64bb14d8f/jaxlib-0.1.64-cp39-none-macosx_10_9_x86_64.whl#sha256=d960b06d1865073228a1e9c11968be4bcd5e383b6d25c02aea014507b2426a73 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Found link https://files.pythonhosted.org/packages/0f/90/ffb9c31876ad73147a107c1299e8d9bb8e0967ac3af8ecf800938b24c61c/jaxlib-0.1.64-cp39-none-manylinux2010_x86_64.whl#sha256=d7674c89980230299caafbccaf8b29c122a4d2f9240eee39d2f4a3e3fadc1e83 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6), version: 0.1.64
  Skipping link: none of the wheel's tags (cp36-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/cd/c6/9d39c16ac49c499e2f57414e59f1deb4fe23dc577a113e1a2eb8feca6ace/jaxlib-0.1.65-cp36-none-macosx_10_9_x86_64.whl#sha256=7a8321fd12b917fe9b9ee21bc6d503739a089f72f45e86137adb21d436bf76d6 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp36-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/42/f0/8dcc0d0bc0de73031ab096a53f64c1fcdfcfc1d4fa77fb20818c4c93201f/jaxlib-0.1.65-cp36-none-manylinux2010_x86_64.whl#sha256=b66c870083dfddff29c929bcd0c09c9f7e059cffe7c87f2e88508974a7a30c5d (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/0c/d7/def00394cd51d9a0697ed1eb808170bc36427173e14f0dea05106c39391b/jaxlib-0.1.65-cp37-none-macosx_10_9_x86_64.whl#sha256=465542b15b30509eef912c75f39c8e51efe815e5f339df91130eb48143432c62 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/bb/64/927d770b40ddc2e247c8fe8823266565fe9d5cd048cca2aa0771baa6ed29/jaxlib-0.1.65-cp37-none-manylinux2010_x86_64.whl#sha256=1080ba936fa3126419966eb93c983b96aa14f149b9fc86928c5583f378d72fb6 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/f8/37/97e1debd30fd55208056f97d15d7d114fa426646a30027d6780dd77a7b47/jaxlib-0.1.65-cp38-none-macosx_10_9_x86_64.whl#sha256=d0e8bf61cdd532e7d4eafa5963d62f15459e4e4f3e46b01f9181f7e9241f7e67 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/b4/4a/3c1a1dc6dd01d45f53fb9afd59a909e1c744bfc7d359309efad12451c7a1/jaxlib-0.1.65-cp38-none-manylinux2010_x86_64.whl#sha256=0b02ac78b3c5b8d8cad6224024539e69e35263b2aa45ec591690fccb729e15a6 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp39-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/53/04/19bb54081726e3109e908cf894160532420bb669e2f6bc716ab246719a71/jaxlib-0.1.65-cp39-none-macosx_10_9_x86_64.whl#sha256=bb73198e6d6b4f9e9d9132f604923c86ce9828af2933476b3c9ee66ce35ab6d6 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Found link https://files.pythonhosted.org/packages/62/9a/57707c563171c1b6ff1a9d4884ea31bd223ae5f18b0eacf465208953e58c/jaxlib-0.1.65-cp39-none-manylinux2010_x86_64.whl#sha256=33b744366e039f9420bf9e73cb3367e820ee9702131413bed4b729b2602dc79b (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6), version: 0.1.65
  Skipping link: none of the wheel's tags (cp36-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/cb/37/c97e268b0e6fd646fd39e32420bda1c66b0d50bd5bfc36e8139475c753cd/jaxlib-0.1.66-cp36-none-macosx_10_9_x86_64.whl#sha256=6b9b935bbf2c285e119a0d8caecf6a2cb3b54aaa1b589a53af959c370d1796b8 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp36-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/50/95/f4aa6c3f4b7d295a5cf0546d6d5a72503c76593d5165d28527fd527febaf/jaxlib-0.1.66-cp36-none-manylinux2010_x86_64.whl#sha256=360523523191c6a11737eb9d064255be7e56cf137fa622d02b31676f446d0bb0 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/36/c5/5886418d675e05aa74dd8cbd35a33cd64f725556fbf9c0a455ad6770f6ea/jaxlib-0.1.66-cp37-none-macosx_10_9_x86_64.whl#sha256=894ea71c99500e238f1745746f5aa2483ed81721c67b3e8ab01090baf6d111c6 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/67/90/a82bd6e9fc2f5c84507d88fa62846a50511299a194edf48630ec4be544a0/jaxlib-0.1.66-cp37-none-manylinux2010_x86_64.whl#sha256=785b7ccbbda4dbc867b87cdbe1f915620788a78609d80771c8a98ad2863c8717 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/cf/8d/0ffa8ae9388f2777727296ee848b3d89d5b82639af293d93f4e19bf0214a/jaxlib-0.1.66-cp38-none-macosx_10_9_x86_64.whl#sha256=ef84b3e2470c209fe9ea7827b196c1f5c109b80a42d3a53c8e121310792f5efd (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/08/57/411dbe2465bae2d8cc45fb818c8e13e2f613217cd238b9ee1e1cb91de1f1/jaxlib-0.1.66-cp38-none-manylinux2010_x86_64.whl#sha256=d47b686a9c799c18a99699632e1754d6f8d92914a6072d9e743b1c4f49a624fb (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp39-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/3e/7c/5caad919bc6b9d6f4bac0bd51fec5657b907d0e7c924d3379f857614e938/jaxlib-0.1.66-cp39-none-macosx_10_9_x86_64.whl#sha256=bfd6d3ad1444d5d4f01550e49f37298d96d808c26017fe90b6f40bbbb7306153 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Found link https://files.pythonhosted.org/packages/fd/9e/ed2f95eae56cb52d1040b483b546774f9382a4feb25852db3a3d55b83de5/jaxlib-0.1.66-cp39-none-manylinux2010_x86_64.whl#sha256=0a0b75edb0b61fbf65d419588b1ea039f699972fd82dc59d015bdc94f2f7b7eb (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6), version: 0.1.66
  Skipping link: none of the wheel's tags (cp36-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/51/0d/2c4fa01f2abee84e04b643eabf7fadaa8d80d7d5777556f0e569f07ec491/jaxlib-0.1.67-cp36-none-macosx_10_9_x86_64.whl#sha256=13420851cb5245ed59f355d2b3f058f4725a421f7f4459e15eed549eafc16064 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp36-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/c5/fd/021a45b9f26bdc50792a13b1c60904de9b2c16f4b2b28accf456f5cdf68e/jaxlib-0.1.67-cp36-none-manylinux2010_x86_64.whl#sha256=0f83f307f3d5864a05dca7a70a667c4c7e7de3cdd5c116b1ec78fc18b6da57cc (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/f3/f3/bb2c7d1cdb20cf5e1923d1732cbe97d1fb7f27c6b1885676c35b3b72b8a1/jaxlib-0.1.67-cp37-none-macosx_10_9_x86_64.whl#sha256=84b2d19d36d6b55e4a8a22a8be7e8e3b67592e808d181485b04f0ce9b27bd2a1 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/dc/0d/ebace284066b58b380316a6211d8e054a027d5d8c9f91170c80313a277f0/jaxlib-0.1.67-cp37-none-manylinux2010_x86_64.whl#sha256=35db3764e6d897ebf4206f904a7febfa2a867e824e219d5e9b16da07192e2ace (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/17/88/b21e75150914df413eaa6d46a61a5232c56d2297ff4ed6a704d835d25ea5/jaxlib-0.1.67-cp38-none-macosx_10_9_x86_64.whl#sha256=94317ffc3df58a27917232119e9103badc90140a40335cebda20a58d8952e76b (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/15/cb/f11daa8be2374351c4a7d6fbcfc9b266950c47c44f5d3f5cb66799b126a8/jaxlib-0.1.67-cp38-none-manylinux2010_x86_64.whl#sha256=961d058fd2e63b093f732575272fa4bcb1a933806a1b03faf7e340e9462fa680 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp39-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/08/95/8d0c91cb0ce78c0aa77b178ebea810e8ec34197875bc4ae42e2e08c822d4/jaxlib-0.1.67-cp39-none-macosx_10_9_x86_64.whl#sha256=fa38b0a5824c230b77f7cd5322e85a59f7822d80b76d751b5c8df4e179e5a701 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Found link https://files.pythonhosted.org/packages/20/84/daab03fbe3bea931ddbbb3d465e07b14121a8f8aac7e11e40093f5f41b74/jaxlib-0.1.67-cp39-none-manylinux2010_x86_64.whl#sha256=aef419f9fb7a02c5892ac0c3f8883fbca6cef480f6836b557943e9b75763d811 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6), version: 0.1.67
  Skipping link: none of the wheel's tags (cp36-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/33/28/038af5145f58ede7ab26bc72d2ccf40adb79cd77831d376979da29f795f1/jaxlib-0.1.68-cp36-none-macosx_10_9_x86_64.whl#sha256=e2495f22d43736f3cc603442d0533bb620904ea24c31242e069705ab885b7db1 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp36-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/dc/6b/6bbfae7c4899ec29e14ee91756c857b6eb8cb38f807e150f8ded9feec544/jaxlib-0.1.68-cp36-none-manylinux2010_x86_64.whl#sha256=2ab48b59170682c8a11018c8b6690f36e5bc27a02d76a82d8c6ee540b5f0f949 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/83/27/ed483e29d210ad4f13887c15e202aed25c7dfe662ee702fe925814777f80/jaxlib-0.1.68-cp37-none-macosx_10_9_x86_64.whl#sha256=c37141512bc60fbc876770c2f01ffbce3480e6ba02836eccf7c15889ad37b47b (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/1f/4c/e72a5450c1f9feefc5eb1927b3a2444dee5c74160f0f02173602abeb3f68/jaxlib-0.1.68-cp37-none-manylinux2010_x86_64.whl#sha256=3c82523a806c36e22583f1785783a7a3927d8f27d45540af41bd37151c417fda (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/e5/1a/6cc61aed1d92f46dca78fb93287e2aa992d9f01692c5009b56e232eb369c/jaxlib-0.1.68-cp38-none-macosx_10_9_x86_64.whl#sha256=2e8c86ea5f8ea38491a69832ef1152f8fdc8136300d321682cb1290ad1a35a70 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/0a/9e/90ff368ff75ac89f1de63359df56f76e7805bfd2626cdbd7f1440ec4b80f/jaxlib-0.1.68-cp38-none-manylinux2010_x86_64.whl#sha256=dac5f5732343f1609e15bbb7a40462bfe95d238f144046ffab2ca3dbb1a21aec (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp39-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/bb/95/d133737e6d02c52e1b6f3698c9aeb5248b62416537c5499f74e37d1cfe25/jaxlib-0.1.68-cp39-none-macosx_10_9_x86_64.whl#sha256=7a9631849a777a8a6206151c8ded3adac70d657dfd67115d75664f669fe7b9a1 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Found link https://files.pythonhosted.org/packages/ab/fd/8856e4b1b1063f2836d50af46cee09bfda422b8af8d357a6783d13232162/jaxlib-0.1.68-cp39-none-manylinux2010_x86_64.whl#sha256=286f938cc7803a6942e4bb6b88a3d7c86977a906e517d804df5af8dc1d15026c (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6), version: 0.1.68
  Skipping link: none of the wheel's tags (cp36-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/8f/f5/8bf33f2b3b0097692df33bbfd36e3350e32c2650ad09c417f253e8e3d00c/jaxlib-0.1.69-cp36-none-macosx_10_9_x86_64.whl#sha256=f40ef7dd6bef4273f68fe5e724f29d0504ccc6b47b31ea19fab305fc4322a272 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp36-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/5e/c0/f0985c39f153aea438ee985452a9fae7fa7e1ca9b87e5cd1fb0b4e93a7a8/jaxlib-0.1.69-cp36-none-manylinux2010_x86_64.whl#sha256=e41830b8b02a44da36b73bf8dc268c3acc9d85194bb96cbed07f1ab24ce6be53 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/87/fb/897d5b8c7bcadec07f919c804636c26ba668d75ce8aba28728cde4837e9b/jaxlib-0.1.69-cp37-none-macosx_10_9_x86_64.whl#sha256=5bb9da8c55afb2ea4bf30fcb1ae4fdc66785201ef7898b31b4e05724277ec382 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp37-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/00/07/e6a6ef696eaef16116d3281944e000d76f29b00d29a9397a2e31f6b02ff9/jaxlib-0.1.69-cp37-none-manylinux2010_x86_64.whl#sha256=99d4fc1ecf940d8d4c28e54e6e1ff99d57978e7e5a73f2db5bfd4b46ffa210f7 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/b6/2c/be7f10d13edafb71f82a559ff6bc617a8547ec3e0aff8886e172b851091b/jaxlib-0.1.69-cp38-none-macosx_10_9_x86_64.whl#sha256=de78cde68962ff18a5c8f051d6d03aa09a7b910216275fc5f9a8890d2f1909de (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp38-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/83/69/225f35a66af2dc00a1d4d6cfdc4aa0d15b560eaa8df3bc25e3486d7154f0/jaxlib-0.1.69-cp38-none-manylinux2010_x86_64.whl#sha256=05cbafbc5878406f4e79cadf22c3a9db735f4a842d3805fc7d0162cbefa2e166 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Skipping link: none of the wheel's tags (cp39-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/a9/1a/0e74ef15c775d95536d331a220e1dc30279b8cb753b0691a64a821f1b108/jaxlib-0.1.69-cp39-none-macosx_10_9_x86_64.whl#sha256=4ad5c340b4f1a47011a8d499aa8590796532c2946691ca414c51e116223cf576 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6)
  Found link https://files.pythonhosted.org/packages/e1/6c/068b2fca04cb9a6f03c12c5d434aaa2c9ba9652d5f79616b3af0f0eafe44/jaxlib-0.1.69-cp39-none-manylinux2010_x86_64.whl#sha256=05858e34e542ed9fd54b066efa9a53ac6297eb5d6c1ee83f67808fc4d17a12bc (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.6), version: 0.1.69
  Skipping link: none of the wheel's tags (cp37-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/52/1e/bef5ac1e48b2c928f5d5967903d3ea76a9e6d4146653f643ff34436ce6ef/jaxlib-0.1.70-cp37-none-macosx_10_9_x86_64.whl#sha256=c2d69253153f2a5afe947cba3d927815290cba7c3b63c096bccbcfdb785251f5 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.7)
  Skipping link: none of the wheel's tags (cp37-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/f2/9c/8fd0cfaa651a4d2697f77e2789d6e0289febe0e86d9558346d200733c3aa/jaxlib-0.1.70-cp37-none-manylinux2010_x86_64.whl#sha256=61a14a13f1279c9febd74aae572feb247ef076656165ba2eb15f4ab204009843 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.7)
  Skipping link: none of the wheel's tags (cp38-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/b4/13/8be2d33410f3b8a3b8898929e73297386dd728606dc706d51008bc9baeb9/jaxlib-0.1.70-cp38-none-macosx_10_9_x86_64.whl#sha256=470e381d77de3dc3fd4a4557f73266212076e461aa2444347ac55dc195369f46 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.7)
  Skipping link: none of the wheel's tags (cp38-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/d3/ea/02192aede22cec707502214e0d89e51f06161e21752c12e92601e4fdb3c9/jaxlib-0.1.70-cp38-none-manylinux2010_x86_64.whl#sha256=89570d3215f5e641d975568b99279b1399e638b5287353d35d2f344ac1472708 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.7)
  Skipping link: none of the wheel's tags (cp39-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/52/93/a1f104b7e312cf44284284259cc66928aceb09483b86b6d5a3d02c16ab23/jaxlib-0.1.70-cp39-none-macosx_10_9_x86_64.whl#sha256=a513d782860ed67fee1657c66e452579b6024a3a05bd5ba684d44a80095df250 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.7)
  Found link https://files.pythonhosted.org/packages/e1/68/e1588ba2bc4b92314e430f3bfef4f575041b0c1d9909a90bb731a0edc901/jaxlib-0.1.70-cp39-none-manylinux2010_x86_64.whl#sha256=9b2b4c4eaa1abe9b916b08f9d8c539fe2163dbb486966f036a8439bddff76ecb (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.7), version: 0.1.70
  Skipping link: none of the wheel's tags (cp37-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/3f/96/7cc0841b7170e63edb28238a4a4e3636b13ea3d080f978a3ca8c096ef0ea/jaxlib-0.1.71-cp37-none-macosx_10_9_x86_64.whl#sha256=ad36895ceb68782bb74f0657da127cece06c19ce1e72ad9c660f79bce549618e (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.7)
  Skipping link: none of the wheel's tags (cp37-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/32/15/fadc902f90ef23f549c49129d6ec0474d93305e3501e996d63da2c073a5c/jaxlib-0.1.71-cp37-none-manylinux2010_x86_64.whl#sha256=059eb572121e3f13e3b841c07137a72f3d0aeb76dc6ddf178922a327994f60b8 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.7)
  Skipping link: none of the wheel's tags (cp38-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/40/7c/1693f9dc35907b839f4779b4d920451c5e6632c31a1e37823c8a580cba56/jaxlib-0.1.71-cp38-none-macosx_10_9_x86_64.whl#sha256=c823b65c95aa7b6d8eb4a45bc6389d43c8d6dd20fe0c531345753819c70cff54 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.7)
  Skipping link: none of the wheel's tags (cp38-none-manylinux2010_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/b6/76/c2890cdfe7a92a2f491643bab39882c11100185898a8c0029ed59ffa5634/jaxlib-0.1.71-cp38-none-manylinux2010_x86_64.whl#sha256=8f4447d6053b55bab9565e365b60c80ff186b0cb05407146ea844d328eaba2bf (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.7)
  Skipping link: none of the wheel's tags (cp39-none-macosx_10_9_x86_64) are compatible (run pip debug --verbose to show compatible tags): https://files.pythonhosted.org/packages/60/43/79c6515d9d77b04c246fa7dc65b3dcca74ee6b06c5d407a8957141ae4fc9/jaxlib-0.1.71-cp39-none-macosx_10_9_x86_64.whl#sha256=f5fc1873c25a7b07f9406bcb09540cef6e2b151aa139be52e34cf35f7b8390b0 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.7)
  Found link https://files.pythonhosted.org/packages/ea/fb/d8ea1a467f4489a4efd7f67602a8ec5a086d7bcde85eff69446dc2334bff/jaxlib-0.1.71-cp39-none-manylinux2010_x86_64.whl#sha256=c66a1bdb57934093938fd6e1767216c198bd1a102b8f5636fdb1f9a5b0d11067 (from https://pypi.org/simple/jaxlib/) (requires-python:>=3.7), version: 0.1.71
Skipping link: not a file: https://pypi.org/simple/jaxlib/
Given no hashes to check 0 links for project 'jaxlib': discarding no candidates
ERROR: Could not find a version that satisfies the requirement jaxlib==0.1.47 (from versions: 0.1.60, 0.1.61, 0.1.62, 0.1.63, 0.1.64, 0.1.65, 0.1.66, 0.1.67, 0.1.68, 0.1.69, 0.1.70, 0.1.71)
ERROR: No matching distribution found for jaxlib==0.1.47
Exception information:
Traceback (most recent call last):
  File "/home/raghurams/repos/optimal-battery-charging/venv/lib/python3.9/site-packages/pip/_vendor/resolvelib/resolvers.py", line 341, in resolve
    name, crit = self._merge_into_criterion(r, parent=None)
  File "/home/raghurams/repos/optimal-battery-charging/venv/lib/python3.9/site-packages/pip/_vendor/resolvelib/resolvers.py", line 173, in _merge_into_criterion
    raise RequirementsConflicted(criterion)
pip._vendor.resolvelib.resolvers.RequirementsConflicted: Requirements conflict: SpecifierRequirement('jaxlib==0.1.47')

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/raghurams/repos/optimal-battery-charging/venv/lib/python3.9/site-packages/pip/_internal/resolution/resolvelib/resolver.py", line 127, in resolve
    result = self._result = resolver.resolve(
  File "/home/raghurams/repos/optimal-battery-charging/venv/lib/python3.9/site-packages/pip/_vendor/resolvelib/resolvers.py", line 473, in resolve
    state = resolution.resolve(requirements, max_rounds=max_rounds)
  File "/home/raghurams/repos/optimal-battery-charging/venv/lib/python3.9/site-packages/pip/_vendor/resolvelib/resolvers.py", line 343, in resolve
    raise ResolutionImpossible(e.criterion.information)
pip._vendor.resolvelib.resolvers.ResolutionImpossible: [RequirementInformation(requirement=SpecifierRequirement('jaxlib==0.1.47'), parent=None)]

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/home/raghurams/repos/optimal-battery-charging/venv/lib/python3.9/site-packages/pip/_internal/cli/base_command.py", line 180, in _main
    status = self.run(options, args)
  File "/home/raghurams/repos/optimal-battery-charging/venv/lib/python3.9/site-packages/pip/_internal/cli/req_command.py", line 205, in wrapper
    return func(self, options, args)
  File "/home/raghurams/repos/optimal-battery-charging/venv/lib/python3.9/site-packages/pip/_internal/commands/install.py", line 318, in run
    requirement_set = resolver.resolve(
  File "/home/raghurams/repos/optimal-battery-charging/venv/lib/python3.9/site-packages/pip/_internal/resolution/resolvelib/resolver.py", line 136, in resolve
    raise error from e
pip._internal.exceptions.DistributionNotFound: No matching distribution found for jaxlib==0.1.47
martinjrobins commented 3 years ago

hmmm, you are right, looks like they have removed the old versions of jaxlib. If you really want to use the jax solver, I think you'll have to upgrade to a newer pybamm version. The newer version is not yet on pypi, so you can clone this repository and install it that way. Follow the developer documentation here: https://pybamm.readthedocs.io/en/latest/install/GNU-linux.html#developer-install

raghuramshankar commented 3 years ago

Hi,

Thanks. I was able to install the developer version and the latest jax and jaxlib versions (with gpu) and it seems to be working with some models now.