forked from AI-Hypercomputer/maxtext
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrequirements.txt
More file actions
43 lines (43 loc) · 829 Bytes
/
Copy pathrequirements.txt
File metadata and controls
43 lines (43 loc) · 829 Bytes
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
absl-py
aqtp
array-record
cloud-accelerator-diagnostics
cloud-tpu-diagnostics
datasets>=4.0.0
flax>=0.11.0
gcsfs
google-api-python-client
google-cloud-aiplatform==1.61.0
google-cloud-monitoring
google-jetstream@git+https://github.com/AI-Hypercomputer/JetStream.git
grain[parquet]>=0.2.6
huggingface_hub
jax>=0.4.30
jaxlib>=0.4.30
jaxtyping
jsonlines
ml-collections
ml-goodput-measurement
mlperf-logging@git+https://github.com/mlperf/logging.git
numpy
omegaconf
optax
orbax-checkpoint>=0.11.22
pathwaysutils>=0.1.1
pillow>=11.1.0
pre-commit
protobuf==3.20.3
pyink
pylint
pytest
pytype
qwix@git+https://github.com/google/qwix.git
sentencepiece==0.2.0
tensorboard-plugin-profile
tensorboardx>=2.6.2.2
tensorflow-datasets
tensorflow-text>=2.17.0
tensorflow>=2.16.0
tiktoken
transformers
tunix@git+https://github.com/google/tunix.git