cartopy chex colabtools dask dm-haiku dm-tree jraph matplotlib pandas rtree scipy trimesh typing_extensions xarray numpy==1.26.4 jupyter google-cloud google-cloud-vision protobuf