Commit f2460e46 authored by VIGNESHinZONE's avatar VIGNESHinZONE
Browse files

remove jax

parent fd0c85c0

env.common.yml

+0 −2
Original line number Diff line number Diff line
@@ -29,5 +29,3 @@ dependencies: 
    - transformers==4.6.*

    - xgboost==1.*

    - git+https://github.com/samoturk/mol2vec
 
    - git+https://github.com/deepmind/dm-haiku

    - git+git://github.com/deepmind/optax.git

env.cpu.mac.yml

+0 −2
Original line number Diff line number Diff line
@@ -7,5 +7,3 @@ dependencies: 
    - torchvision==0.9.1

    - torch-scatter

    - torch-sparse
 
    - jax==0.2.14

    - jaxlib==0.1.66

env.cpu.yml

+0 −2
Original line number Diff line number Diff line
@@ -7,5 +7,3 @@ dependencies: 
    - torchvision==0.9.1+cpu

    - torch-scatter

    - torch-sparse
 
    - jax==0.2.14

    - jaxlib==0.1.66

env.gpu.yml

+0 −3
Original line number Diff line number Diff line
@@ -2,11 +2,8 @@ dependencies: 
  - pip:

    - -f https://download.pytorch.org/whl/torch_stable.html

    - -f https://pytorch-geometric.com/whl/torch-1.8.1+cu111.html

    - -f https://storage.googleapis.com/jax-releases/jax_releases.html

    - dgl-cu110==0.6.*

    - torch==1.8.1+cu111

    - torchvision==0.9.1+cu111

    - torch-scatter

    - torch-sparse
 
    - jax==0.2.15

    - jaxlib==0.1.67+cuda111

CRA Git | Maintained and supported by SUSTech CRA and CCSE