flax
✍️ Expand the Flax annotated MNIST image classification tutorial, port to the Linen API
#764
Merged
Overview
Commits
9
Changes
View On GitHub

✍️ Expand the Flax annotated MNIST image classification tutorial, port to the Linen API #764

avital merged 9 commits into google:master from annotated
8bitmp3
8bitmp3 Expand the Flax annotated MNIST image classification tutorial, port t…
d7559090
google-cla google-cla added cla: yes
8bitmp3 Add blank lines before lists for better Markdown rendering
69d88c8f
codecov-io
8bitmp3 Remove dataset loading output (unnecessary)
0ade549f
8bitmp3 Add a reference to the Linen intro notebook
4ebd3349
8bitmp3 Refactor jit explanation, use parameters instead of weights in text
db5b63c0
8bitmp3 Add PRNG design notes to help new JAX users, refactor the train_epoch…
88011afe
8bitmp3 Use jax.nn.one_hot instead of a custom one-hot encoder
3110203b
8bitmp3 Remove installation output, fix formatting
a9277c07
avital
avital commented on 2020-12-31
8bitmp3
8bitmp3
8bitmp3 Lighter - make annotated MNIST image classification example with Flax…
feefec92
8bitmp3
andsteing andsteing assigned avital avital 5 years ago
BertrandRdp BertrandRdp added Priority: P2 - no schedule
avital
avital avital added pull ready
avital avital merged dcbe4721 into master 5 years ago
8bitmp3 8bitmp3 deleted the annotated branch 5 years ago
avital

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
cla: yes Priority: P2 - no schedule pull ready
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum