├── .gitignore ├── LICENSE ├── README.md ├── assets ├── .DS_Store ├── benchmark.png └── matmul_transpose_kernel.png ├── benchmark.py ├── flash_muon ├── __init__.py ├── matmul_transpose_triton.py └── muon.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nil0x9/flash-muon/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nil0x9/flash-muon/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nil0x9/flash-muon/HEAD/README.md -------------------------------------------------------------------------------- /assets/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nil0x9/flash-muon/HEAD/assets/.DS_Store -------------------------------------------------------------------------------- /assets/benchmark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nil0x9/flash-muon/HEAD/assets/benchmark.png -------------------------------------------------------------------------------- /assets/matmul_transpose_kernel.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nil0x9/flash-muon/HEAD/assets/matmul_transpose_kernel.png -------------------------------------------------------------------------------- /benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nil0x9/flash-muon/HEAD/benchmark.py -------------------------------------------------------------------------------- /flash_muon/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nil0x9/flash-muon/HEAD/flash_muon/__init__.py -------------------------------------------------------------------------------- /flash_muon/matmul_transpose_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nil0x9/flash-muon/HEAD/flash_muon/matmul_transpose_triton.py -------------------------------------------------------------------------------- /flash_muon/muon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nil0x9/flash-muon/HEAD/flash_muon/muon.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nil0x9/flash-muon/HEAD/setup.py --------------------------------------------------------------------------------