Could you please share requirements.txt?

[chinembedded]

HI, Thank you for this example notebook. It's amazing. Could you please share requirements.txt as I am facing some version issues with numpy.

[eamid]

Here are the requirements for TriMap in JAX: https://github.com/google-research/google-research/blob/master/trimap/requirements.txt

To run the colab, make sure you have the rest of the packages installed:

matplotlib
scipy
umap
keras
tensorflow_datasets

Please let me know if you're still having trouble.