search

berenslab / RFEst

A Python 3 toolbox for neural receptive field estimation using splines and Gaussian priors.
https://arxiv.org/abs/2108.07537
GNU General Public License v3.0
24 stars 3 forks source link

Jax lib version not available anymore #6

Closed marcbue closed 2 years ago

marcbue commented 2 years ago

Hi,

I wanted to try out the package but unfortunately the jaxlib version specified in the requirements is not available on Pypi anymore. using a newer one leads to an error that a function is missing in jaxlib.

huangziwei commented 2 years ago

Hi.

I just removed the versions restriction for both Jax and jaxlib.

They were restricted to older versions because a bug in the newer versions that make the computation significantly slower on the CPU. I just tested with the latest versions of them and now it works fine.