google-deepmind / google-deepmind/neural_testbed
Dependencies
Open
- Dominant language
- Jupyter Notebook
- Stars
- 192
- Forks
- 22
- PR merge metrics
- No merged PRs in 30d
Description
Hi There,
Thanks for open-sourcing!
The dependencies are really hard to reconstruct as it seems that Jax is changing its API quite heavily between minor releases.
How to reproduce the bug: Follow your installation manual (in my case with Python 3.9). It will install Jax 0.2 which does not work with your code :( I got it to work by changing the Jax version to 0.1.71 on the CPU but for CUDA I didn't have luck yet.
(Sorry for the tone wrote this after trying to get the dependencies right for half an hour..)
Contributor guide
Assessment
This issue has not been assessed yet.