Skip to content

Conversation

@amrzv
Copy link

@amrzv amrzv commented Jan 13, 2023

Fixed #1 : changed deprecated jax.tree_multimap to jax.tree_map
Also prettified notebook according to pep8.

Tested in colab: "Run all cells" without errors.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

AttributeError in 01_intro.ipynb notebook

1 participant