flax
Update Flax NNX vs JAX Transformations guide
#4286

Merged

Update Flax NNX vs JAX Transformations guide #4286

copybara-service merged 1 commit into google:main from 8bitmp3:update-nnx-jax-transforms

8bitmp3 requested a review from

cgarciae 1 year ago

8bitmp3 requested a review from

IvyZX 1 year ago

8bitmp3 requested a review from

chiamp 1 year ago

8bitmp3 assigned

8bitmp3 1 year ago

8bitmp3 removed review request from

chiamp 1 year ago

IvyZX approved these changes on 2024-12-11

Update Flax NNX vs JAX Transformations guide

c8bfd2a6

8bitmp3 force pushed to c8bfd2a6 1 year ago

8bitmp3 added pull ready

copybara-service merged 207966e5 into main 1 year ago

Reviewers

IvyZX

cgarciae

Assignees

8bitmp3

Labels

pull ready

Milestone

No milestone