-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathrequirements-dev.lock
130 lines (129 loc) · 2.43 KB
/
requirements-dev.lock
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
# generated by rye
# use `rye lock` or `rye sync` to update this lockfile
#
# last locked with the following flags:
# pre: false
# features: []
# all-features: false
# with-sources: false
-e file:.
asttokens==2.4.1
# via icecream
beartype==0.18.5
# via ndpatchify
# via plum-dispatch
colorama==0.4.6
# via icecream
executing==2.0.1
# via icecream
filelock==3.14.0
# via torch
# via triton
fsspec==2024.3.1
# via torch
icecream==2.1.3
iniconfig==2.0.0
# via pytest
jax==0.4.26
# via ndpatchify
jaxlib==0.4.26
# via jax
jaxtyping==0.2.28
# via ndpatchify
jinja2==3.1.3
# via torch
markdown-it-py==3.0.0
# via rich
markupsafe==2.1.5
# via jinja2
mdurl==0.1.2
# via markdown-it-py
ml-dtypes==0.4.0
# via jax
# via jaxlib
mpmath==1.3.0
# via sympy
networkx==3.3
# via torch
numpy==1.26.4
# via jax
# via jaxlib
# via jaxtyping
# via ml-dtypes
# via ndpatchify
# via opt-einsum
# via pandas
# via scipy
# via xarray
nvidia-cublas-cu12==12.1.3.1
# via nvidia-cudnn-cu12
# via nvidia-cusolver-cu12
# via torch
nvidia-cuda-cupti-cu12==12.1.105
# via torch
nvidia-cuda-nvrtc-cu12==12.1.105
# via torch
nvidia-cuda-runtime-cu12==12.1.105
# via torch
nvidia-cudnn-cu12==8.9.2.26
# via torch
nvidia-cufft-cu12==11.0.2.54
# via torch
nvidia-curand-cu12==10.3.2.106
# via torch
nvidia-cusolver-cu12==11.4.5.107
# via torch
nvidia-cusparse-cu12==12.1.0.106
# via nvidia-cusolver-cu12
# via torch
nvidia-nccl-cu12==2.20.5
# via torch
nvidia-nvjitlink-cu12==12.4.127
# via nvidia-cusolver-cu12
# via nvidia-cusparse-cu12
nvidia-nvtx-cu12==12.1.105
# via torch
opt-einsum==3.3.0
# via jax
optree==0.11.0
# via ndpatchify
packaging==24.0
# via pytest
# via xarray
pandas==2.2.2
# via xarray
pluggy==1.5.0
# via pytest
plum-dispatch==2.3.5
# via ndpatchify
pygments==2.17.2
# via icecream
# via rich
pytest==8.2.0
python-dateutil==2.9.0.post0
# via pandas
pytz==2024.1
# via pandas
rich==13.7.1
# via plum-dispatch
scipy==1.13.0
# via jax
# via jaxlib
six==1.16.0
# via asttokens
# via python-dateutil
sympy==1.12
# via torch
torch==2.3.0
# via ndpatchify
triton==2.3.0
# via torch
typeguard==2.13.3
# via jaxtyping
typing-extensions==4.11.0
# via optree
# via torch
tzdata==2024.1
# via pandas
xarray==2024.3.0
# via ndpatchify