FractalSig/environment.yaml at main · javierdejesusda/FractalSig · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
name: fractalsig
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults
dependencies:
  - python=3.13
  - pip
  - pytorch=2.5.1
  - pytorch-cuda=12.4
  - cudatoolkit
  - numpy=2.2.2
  - matplotlib=3.10.0
  - scipy=1.15.1
  - pyyaml=6.0.2
  - tqdm=4.67.1
  - pillow=12.0.0
  - scikit-learn=1.6.1
  - sympy=1.13.1
  - pip:
    - --no-build-isolation
    - iisignature==0.24
    - jax[cuda12]==0.4.38
    - jaxlib==0.4.38
    - jax-cuda12-pjrt==0.4.38
    - jax-cuda12-plugin==0.4.38
    - diffrax==0.6.2
    - equinox==0.11.11
    - lineax==0.0.8
    - optimistix==0.0.11
    - jaxtyping==0.2.36
    - optax==0.2.4
    - chex==0.1.90
    - hydra-core==1.3.2
    - omegaconf==2.3.0
    - fbm==0.3.0
    - ptwt==1.0.1
    - pywavelets==1.9.0
    - typeguard==2.13.3
    - torchdiffeq==0.2.5
    - scienceplots
    - pyparsing==3.2.1
    - six==1.17.0