flax is significantly slower than pytorch
Open
- Dominant language
- Jupyter Notebook
- Stars
- 7.3k
- Forks
- 833
- Avg merge
- 5h 11m
- Merged PRs (30d)
- 5
Description
minist classification using convolutional networks
link: https://flax.readthedocs.io/en/latest/quick_start.html
same approach on pytorch is significantly faster, (almost 5 times faster)
flax version takes 50 seconds per epoch!!!
Contributor guide
Assessment
This issue has not been assessed yet.