Skip to content
#

llm-efficiency

Here are 3 public repositories matching this topic...

UnSwag is a memory-efficient training primitive for the JAX/TPU and PyTorch/GPU ecosystems. By mapping ReLU activations to 1-bit structural isomorphisms, UnSwag reduces activation memory by 32x with 0.000000 loss difference. - TPU Mode: Uses JAX/Pallas for massive context windows on Google TPUs. The Memory Wall is now optional.

  • Updated Dec 21, 2025
  • Python

Improve this page

Add a description, image, and links to the llm-efficiency topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the llm-efficiency topic, visit your repo's landing page and select "manage topics."

Learn more