-
Notifications
You must be signed in to change notification settings - Fork 40
/
requirements.in
27 lines (26 loc) · 693 Bytes
/
requirements.in
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
# To update requirements.txt for praxis and paxml, run:
# cd ../../paxml/pip_package && bash ./compile_requirements.sh
absl-py
# enforce newer chex version to avoid deprecated jax syntax
chex>=0.1.85
clu @ git+https://github.com/google/CommonLoopUtils
einops
etils
fiddle @ git+https://github.com/google/fiddle
flax @ git+https://github.com/google/flax
jax @ git+https://github.com/google/jax
jax-bitempered-loss
jaxtyping
lingvo
numpy<2
optax
optax-shampoo
opt-einsum
# ensure sentencepiece is compataible with protobuf 3.19.6
sentencepiece==0.1.99
tensorflow-datasets==4.8.3
tensorflow-metadata==1.12.0
tensorflow-text~=2.9.0
tensorflow~=2.9.2
tfds-nightly==4.8.3.dev202303280045
typeguard