flax
Add Flax FAQ - how to search, how to take the derivative w.r.t. a hidden layer, remat_scan vs scan(remat), recommended training libraries/metrics
#3267

Merged

Add Flax FAQ - how to search, how to take the derivative w.r.t. a hidden layer, remat_scan vs scan(remat), recommended training libraries/metrics #3267

copybara-service merged 2 commits into google:main from 8bitmp3:flax-faq

8bitmp3 requested a review from

chiamp 2 years ago

8bitmp3 assigned

8bitmp3 2 years ago

8bitmp3 requested a review from

marcvanzee 2 years ago

8bitmp3 requested a review from

IvyZX 2 years ago

8bitmp3 changed the title ~~Add Flax FAQ - how to search, how to take a derivative w.r.t. a hidden layer~~ Add Flax FAQ - how to search, how to take a derivative w.r.t. a hidden layer, remat_scan vs scan(remat) 2 years ago

8bitmp3 requested a review from

cgarciae 2 years ago

8bitmp3 force pushed 2 years ago

8bitmp3 changed the title ~~Add Flax FAQ - how to search, how to take a derivative w.r.t. a hidden layer, remat_scan vs scan(remat)~~ Add Flax FAQ - how to search, how to take the derivative w.r.t. a hidden layer, remat_scan vs scan(remat) 2 years ago

cgarciae commented on 2023-08-15

chiamp commented on 2023-08-22

chiamp commented on 2023-10-09

8bitmp3 changed the title ~~Add Flax FAQ - how to search, how to take the derivative w.r.t. a hidden layer, remat_scan vs scan(remat)~~ Add Flax FAQ - how to search, how to take the derivative w.r.t. a hidden layer, remat_scan vs scan(remat), recommended training libraries/metrics 2 years ago

8bitmp3 force pushed 2 years ago

8bitmp3 closed this 2 years ago

8bitmp3 force pushed to 66354365 2 years ago

Add Flax F.A.Q.

d0216d31

8bitmp3 reopened this 2 years ago

Update Flax F.A.Q.

7b1d8a61

chiamp approved these changes on 2023-10-09

chiamp added pull ready

copybara-service merged 30147a4c into main 2 years ago

8bitmp3 deleted the flax-faq branch 2 years ago

Reviewers

chiamp

cgarciae

marcvanzee

IvyZX

Assignees

8bitmp3

Labels

pull ready

Milestone

No milestone

flax Add Flax FAQ - how to search, how to take the derivative w.r.t. a hidden layer, remat_scan vs scan(remat), recommended training libraries/metrics #3267 Merged

Add Flax FAQ - how to search, how to take the derivative w.r.t. a hidden layer, remat_scan vs scan(remat), recommended training libraries/metrics #3267

flax
Add Flax FAQ - how to search, how to take the derivative w.r.t. a hidden layer, remat_scan vs scan(remat), recommended training libraries/metrics
#3267

Merged