Skip to content

Instantly share code, notes, and snippets.

@NobuoTsukamoto

NobuoTsukamoto/flax-nnx-model-ema.ipynb

Created May 11, 2025 22:50
Show Gist options
  • Save NobuoTsukamoto/61eb086f4bf1230c9613a3035dae8cec to your computer and use it in GitHub Desktop.
Save NobuoTsukamoto/61eb086f4bf1230c9613a3035dae8cec to your computer and use it in GitHub Desktop.
Download ZIP
Flax NNX Model EMA.ipynb
Display the source blob
Display the rendered blob
Raw
flax-nnx-model-ema.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment