Skip to content

GitLab

Commit 7c75b98e authored by suily's avatar suily
Browse files

Update requirements.txt

parent cf86f723
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
requirements.txt
View file @ 7c75b98e
...@@ -3,7 +3,6 @@ absl-py>=0.12.0 ...@@ -3,7 +3,6 @@ absl-py>=0.12.0
chex>=0.0.7 chex>=0.0.7
clu>=0.0.3 clu>=0.0.3
einops>=0.3.0 einops>=0.3.0
flax>=0.6.4
git+https://github.com/google/flaxformer git+https://github.com/google/flaxformer
# jax[cuda11_cudnn86]>=0.4.2 # jax[cuda11_cudnn86]>=0.4.2
#--find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html #--find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment