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

Update requirements.txt

parent cf86f723
......@@ -3,7 +3,6 @@ absl-py>=0.12.0
chex>=0.0.7
clu>=0.0.3
einops>=0.3.0
flax>=0.6.4
git+https://github.com/google/flaxformer
# jax[cuda11_cudnn86]>=0.4.2
#--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