Tweeted By @DynamicWebPaige
✨🤯 It still boggles my mind how concise deep learning with #JAX can be.
— 👩💻 DynamicWebPaige (@DynamicWebPaige) February 21, 2020
Check out this *full example, with hyperparameters* of training a ResNet50 on ImageNet2012, using @DeepMind's high-level API, Haiku:https://t.co/nNUsaJwqiC
Note: 76.2-76.4% top-1 at batch size 1024. pic.twitter.com/SpRnNi4SPs