Skip to content

Instantly share code, notes, and snippets.

@LiutongZhou

LiutongZhou/JAX Sharding Cheat Sheet (v0.8+).ipynb

Last active March 19, 2026 02:30
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save LiutongZhou/139eddad3ea732b9c5f06b97cf702799 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save LiutongZhou/139eddad3ea732b9c5f06b97cf702799 to your computer and use it in GitHub Desktop.
Download ZIP
Jax Distributed Zero to Hero
Display the source blob
Display the rendered blob
Raw
JAX Sharding Cheat Sheet (v0.8+).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