google-deepmind / google-deepmind/educational
pin the versions for the library installations
Open
- Dominant language
- Jupyter Notebook
- Stars
- 1.5k
- Forks
- 203
- PR merge metrics
- No merged PRs in 30d
Description
For the Introduction to Graph Neural Nets with JAX/jraph notebook in colab, I get an error `AttributeError: module 'jaxlib.xla_extension' has no attribute 'PmapFunction'`.
I didn't get this before but I'm guessing it's because the versions of the pip installs have changed. Would it be possible to pin the library versions in cell 1 so that the notebook always work even a year from now?
Contributor guide
Assessment
This issue has not been assessed yet.