complex-ema

Here is 1 public repository matching this topic...

Pure PyTorch + 🤗 Transformers reimplementation of Megalodon (CEMA + chunked attention) - readable, hackable, no CUDA kernels required

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

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