Skip to content

Commit

Permalink
Adds requirements file for AQT imagenet.
Browse files Browse the repository at this point in the history 
PiperOrigin-RevId: 414111053
  • Loading branch information
@shivaniag @copybara-github
shivaniag authored and copybara-github committed Dec 4, 2021
1 parent c96717a commit 7cdec61
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions aqt/jax/imagenet/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
dacite
jax
ml-collections
tensorflow
tensorflow-datasets

0 comments on commit 7cdec61

Please sign in to comment.