Commit f219c2b9 authored by VIGNESHinZONE's avatar VIGNESHinZONE
Browse files

removing jax dependencies

parent 86d025b2

env.common.yml

+0 −2
Original line number Diff line number Diff line
@@ -29,6 +29,4 @@ 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 −1
Original line number Diff line number Diff line
@@ -7,4 +7,3 @@ dependencies: 
    - torchvision==0.9.1

    - torch-scatter

    - torch-sparse
 
    - jax[cpu]

env.cpu.yml

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

    - torch-scatter

    - torch-sparse
 
    - jax[cpu]

env.gpu.yml

+0 −1
Original line number Diff line number Diff line
@@ -7,5 +7,4 @@ dependencies: 
    - torchvision==0.9.1+cu111

    - torch-scatter

    - torch-sparse

    - jax[cuda111]

CRA Git | Maintained and supported by SUSTech CRA and CCSE