Dependency issue

[shenvitor]

What happened?

TR-000 run in Colab is not successful due to the following issues.

Relevant log output

Is it possible to reproduce the bug with a small code snippet?

See above

Additional steps to reproduce the bug

No

Which Python version were you using?

None

Python dependencies

No response

[redeboer]

Thanks for posting! Resolving the conflict for this notebook should be easy, but it does point to a larger problem with the dependency 'pinning' in the technical reports (see also #256, and indirectly the discussion in #151 (comment)). The ideal would be to pin all dependencies instead of only the direct ones.

@shenvitor can you fix the notebook for this bug by finding a jaxlib version that works and pinning to that? A solution should be found for the larger issue here, but I'll describe that in a different issue.

[shenvitor]

Reference to this related problem: Dependency Resolution