Loading env.common.yml +0 −2 Original line number Original line Diff line number Diff line Loading @@ -25,8 +25,6 @@ dependencies: - tensorflow~=2.4 - tensorflow~=2.4 - tensorflow_probability==0.12.* - tensorflow_probability==0.12.* - tensorflow_addons==0.12.* - tensorflow_addons==0.12.* - jax==0.2.14 - jaxlib==0.1.67 - dm-haiku==0.0.4 - dm-haiku==0.0.4 - optax==0.0.8 - optax==0.0.8 - torch-geometric==1.7.0 - torch-geometric==1.7.0 Loading env.cpu.mac.yml +2 −1 Original line number Original line Diff line number Diff line Loading @@ -7,3 +7,4 @@ dependencies: - torchvision==0.9.1 - torchvision==0.9.1 - torch-scatter - torch-scatter - torch-sparse - torch-sparse - jax[cpu] env.cpu.yml +2 −1 Original line number Original line Diff line number Diff line Loading @@ -7,3 +7,4 @@ dependencies: - torchvision==0.9.1+cpu - torchvision==0.9.1+cpu - torch-scatter - torch-scatter - torch-sparse - torch-sparse - jax[cpu] env.gpu.yml +2 −1 Original line number Original line Diff line number Diff line Loading @@ -7,3 +7,4 @@ dependencies: - torchvision==0.9.1+cu111 - torchvision==0.9.1+cu111 - torch-scatter - torch-scatter - torch-sparse - torch-sparse - jax[cuda111] Loading
env.common.yml +0 −2 Original line number Original line Diff line number Diff line Loading @@ -25,8 +25,6 @@ dependencies: - tensorflow~=2.4 - tensorflow~=2.4 - tensorflow_probability==0.12.* - tensorflow_probability==0.12.* - tensorflow_addons==0.12.* - tensorflow_addons==0.12.* - jax==0.2.14 - jaxlib==0.1.67 - dm-haiku==0.0.4 - dm-haiku==0.0.4 - optax==0.0.8 - optax==0.0.8 - torch-geometric==1.7.0 - torch-geometric==1.7.0 Loading
env.cpu.mac.yml +2 −1 Original line number Original line Diff line number Diff line Loading @@ -7,3 +7,4 @@ dependencies: - torchvision==0.9.1 - torchvision==0.9.1 - torch-scatter - torch-scatter - torch-sparse - torch-sparse - jax[cpu]
env.cpu.yml +2 −1 Original line number Original line Diff line number Diff line Loading @@ -7,3 +7,4 @@ dependencies: - torchvision==0.9.1+cpu - torchvision==0.9.1+cpu - torch-scatter - torch-scatter - torch-sparse - torch-sparse - jax[cpu]
env.gpu.yml +2 −1 Original line number Original line Diff line number Diff line Loading @@ -7,3 +7,4 @@ dependencies: - torchvision==0.9.1+cu111 - torchvision==0.9.1+cu111 - torch-scatter - torch-scatter - torch-sparse - torch-sparse - jax[cuda111]
