flax icon indicating copy to clipboard operation
flax copied to clipboard
verified publisher icon google

Update Flax NNX vs JAX Transformations guide

Open 8bitmp3 opened this issue 1 year ago • 0 comments

Preview: https://flax--4286.org.readthedocs.build/en/4286/guides/jax_and_nnx_transforms.html

@cgarciae @IvyZX PTAL

8bitmp3 avatar Oct 10 '24 23:10 8bitmp3