Skip to content

Instantly share code, notes, and snippets.

@zhangqiaorjc

zhangqiaorjc/copy-of-jax-transformer-model-for-fp8-no-decode-cache-shared-with-nvidia.ipynb

Created December 14, 2022 07:32
Show Gist options
  • Save zhangqiaorjc/24f1c3dc7674cfd8b0fb588541c94c70 to your computer and use it in GitHub Desktop.
Save zhangqiaorjc/24f1c3dc7674cfd8b0fb588541c94c70 to your computer and use it in GitHub Desktop.
Download ZIP
Copy of JAX Transformer model for fp8 (no decode cache) shared with NVIDIA.ipynb
Display the source blob
Display the rendered blob
Raw
copy-of-jax-transformer-model-for-fp8-no-decode-cache-shared-with-nvidia.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