Skip to content

Instantly share code, notes, and snippets.

@shoyer

shoyer/jax-einsum-primitive.ipynb

Created May 18, 2023 17:27
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save shoyer/57107c2d2023d708683cdb5087810179 to your computer and use it in GitHub Desktop.
Save shoyer/57107c2d2023d708683cdb5087810179 to your computer and use it in GitHub Desktop.
Download ZIP
JAX einsum primitive .ipynb
Display the source blob
Display the rendered blob
Raw
jax-einsum-primitive.ipynb
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