Skip to content

Commit 63aa0eb

Browse files
authored
add jax_tpu_embedding to requirements (#98)
1 parent 0d813eb commit 63aa0eb

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
absl-py
22
clu
33
jax[cpu]
4+
jax_tpu_embedding
45
keras-hub
56
keras-nlp
67
keras-rs

0 commit comments

Comments
 (0)