-
Notifications
You must be signed in to change notification settings - Fork 2
/
requirements.txt
128 lines (128 loc) · 2.36 KB
/
requirements.txt
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
# This file was autogenerated by uv via the following command:
# uv pip compile pyproject.toml -o requirements.txt
absl-py==2.1.0
# via
# chex
# dm-haiku
# ml-collections
# optax
# orbax-checkpoint
chex==0.1.85
# via optax
contextlib2==21.6.0
# via ml-collections
contourpy==1.2.0
# via matplotlib
cycler==0.12.1
# via matplotlib
dm-haiku==0.0.12
einops==0.7.0
etils==1.7.0
# via orbax-checkpoint
flax==0.8.1
# via dm-haiku
fonttools==4.49.0
# via matplotlib
fsspec==2024.2.0
# via etils
importlib-resources==6.1.3
# via etils
jax==0.4.25
# via
# chex
# flax
# optax
# orbax-checkpoint
jaxlib==0.4.25
# via
# chex
# optax
# orbax-checkpoint
jmp==0.0.4
# via dm-haiku
kiwisolver==1.4.5
# via matplotlib
markdown-it-py==3.0.0
# via rich
matplotlib==3.8.3
mdurl==0.1.2
# via markdown-it-py
ml-collections @ git+https://github.com/danielkelshaw/ml_collections@b669f316367406c96d250b16d72f44a730b5b071
ml-dtypes==0.3.2
# via
# jax
# jaxlib
# tensorstore
msgpack==1.0.8
# via
# flax
# orbax-checkpoint
nest-asyncio==1.6.0
# via orbax-checkpoint
numpy==1.26.4
# via
# chex
# contourpy
# dm-haiku
# flax
# jax
# jaxlib
# jmp
# matplotlib
# ml-dtypes
# opt-einsum
# optax
# orbax-checkpoint
# scipy
# tensorstore
opt-einsum @ git+https://github.com/dgasmith/opt_einsum@1a984b7b75f3e532e7129f6aa13f7ddc3da66e10
# via jax
optax==0.2.1
# via flax
orbax-checkpoint==0.5.3
# via flax
packaging==24.0
# via matplotlib
pillow==10.2.0
# via matplotlib
protobuf==4.25.3
# via orbax-checkpoint
pygments==2.17.2
# via rich
pyparsing==3.1.2
# via matplotlib
python-dateutil==2.9.0.post0
# via matplotlib
pyyaml==6.0.1
# via
# flax
# ml-collections
# orbax-checkpoint
rich==13.7.1
# via flax
scipy==1.12.0
# via
# jax
# jaxlib
setuptools==69.1.1
# via chex
six==1.16.0
# via
# ml-collections
# python-dateutil
tabulate==0.9.0
# via dm-haiku
tensorstore==0.1.54
# via
# flax
# orbax-checkpoint
toolz==0.12.1
# via chex
typing-extensions==4.10.0
# via
# chex
# etils
# flax
# orbax-checkpoint
zipp==3.17.0
# via etils