flax

a50e28d8 - Adds tensorflow to imagenet requirements.txt