wassname/evil_MoE
1
0
Fork 0
mirror of https://github.com/wassname/evil_MoE.git synced 2026-06-27 16:15:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

docs

Browse Source
This commit is contained in:
wassname
2026-06-06 12:27:26 +00:00
parent 842a373ebc
commit bcf09dd742
13 changed files with 181 additions and 1113 deletions
Download Patch File Download Diff File Expand all files Collapse all files
uv.lock
Generated
+69 -69
View File
@@ -1061,13 +1061,13 @@ name = "mlx-lm"
version = "0.29.1"
source = { registry = "https://pypi.org/simple" }
dependencies = [
{ name = "jinja2" },
{ name = "jinja2", marker = "sys_platform == 'darwin'" },
{ name = "mlx", marker = "sys_platform == 'darwin'" },
{ name = "numpy" },
{ name = "protobuf" },
{ name = "pyyaml" },
{ name = "sentencepiece" },
{ name = "transformers" },
{ name = "numpy", marker = "sys_platform == 'darwin'" },
{ name = "protobuf", marker = "sys_platform == 'darwin'" },
{ name = "pyyaml", marker = "sys_platform == 'darwin'" },
{ name = "sentencepiece", marker = "sys_platform == 'darwin'" },
{ name = "transformers", marker = "sys_platform == 'darwin'" },
]
sdist = { url = "https://files.pythonhosted.org/packages/e3/62/f46e1355256a114808517947f8e83ad6be310c7288c551db0fa678f47923/mlx_lm-0.29.1.tar.gz", hash = "sha256:b99180d8f33d33a077b814e550bfb2d8a59ae003d668fd1f4b3fff62a381d34b", size = 232302, upload-time = "2025-12-16T16:58:27.959Z" }
wheels = [
@@ -1282,7 +1282,7 @@ name = "nvidia-cudnn-cu12"
version = "9.10.2.21"
source = { registry = "https://pypi.org/simple" }
dependencies = [
{ name = "nvidia-cublas-cu12" },
{ name = "nvidia-cublas-cu12", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')" },
]
wheels = [
{ url = "https://files.pythonhosted.org/packages/ba/51/e123d997aa098c61d029f76663dedbfb9bc8dcf8c60cbd6adbe42f76d049/nvidia_cudnn_cu12-9.10.2.21-py3-none-manylinux_2_27_x86_64.whl", hash = "sha256:949452be657fa16687d0930933f032835951ef0892b37d2d53824d1a84dc97a8", size = 706758467, upload-time = "2025-06-06T21:54:08.597Z" },
@@ -1293,7 +1293,7 @@ name = "nvidia-cufft-cu12"
version = "11.3.3.83"
source = { registry = "https://pypi.org/simple" }
dependencies = [
{ name = "nvidia-nvjitlink-cu12" },
{ name = "nvidia-nvjitlink-cu12", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')" },
]
wheels = [
{ url = "https://files.pythonhosted.org/packages/1f/13/ee4e00f30e676b66ae65b4f08cb5bcbb8392c03f54f2d5413ea99a5d1c80/nvidia_cufft_cu12-11.3.3.83-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:4d2dd21ec0b88cf61b62e6b43564355e5222e4a3fb394cac0db101f2dd0d4f74", size = 193118695, upload-time = "2025-03-07T01:45:27.821Z" },
@@ -1320,9 +1320,9 @@ name = "nvidia-cusolver-cu12"
version = "11.7.3.90"
source = { registry = "https://pypi.org/simple" }
dependencies = [
{ name = "nvidia-cublas-cu12" },
{ name = "nvidia-cusparse-cu12" },
{ name = "nvidia-nvjitlink-cu12" },
{ name = "nvidia-cublas-cu12", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')" },
{ name = "nvidia-cusparse-cu12", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')" },
{ name = "nvidia-nvjitlink-cu12", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')" },
]
wheels = [
{ url = "https://files.pythonhosted.org/packages/85/48/9a13d2975803e8cf2777d5ed57b87a0b6ca2cc795f9a4f59796a910bfb80/nvidia_cusolver_cu12-11.7.3.90-py3-none-manylinux_2_27_x86_64.whl", hash = "sha256:4376c11ad263152bd50ea295c05370360776f8c3427b30991df774f9fb26c450", size = 267506905, upload-time = "2025-03-07T01:47:16.273Z" },
@@ -1333,7 +1333,7 @@ name = "nvidia-cusparse-cu12"
version = "12.5.8.93"
source = { registry = "https://pypi.org/simple" }
dependencies = [
{ name = "nvidia-nvjitlink-cu12" },
{ name = "nvidia-nvjitlink-cu12", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')" },
]
wheels = [
{ url = "https://files.pythonhosted.org/packages/c2/f5/e1854cb2f2bcd4280c44736c93550cc300ff4b8c95ebe370d0aa7d2b473d/nvidia_cusparse_cu12-12.5.8.93-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:1ec05d76bbbd8b61b06a80e1eaf8cf4959c3d4ce8e711b65ebd0443bb0ebb13b", size = 288216466, upload-time = "2025-03-07T01:48:13.779Z" },
@@ -1582,60 +1582,6 @@ wheels = [
{ url = "https://files.pythonhosted.org/packages/a3/da/30f15f0c3959b70e7a6583eccd37140a30b5c643ca374792d150b3a357df/polars_runtime_32-1.41.0-cp310-abi3-win_arm64.whl", hash = "sha256:ed922400f0eb393345fd7b6874b150eb943af2b816297a3dde03735cb5f3de08", size = 45921961, upload-time = "2026-05-22T20:19:59.525Z" },
]
[[package]]
name = "projected-grpo"
version = "0.1.0"
source = { editable = "." }
dependencies = [
{ name = "beartype" },
{ name = "causal-conv1d" },
{ name = "datasets" },
{ name = "einops" },
{ name = "flash-attn" },
{ name = "flash-linear-attention" },
{ name = "huggingface-hub" },
{ name = "jaxtyping" },
{ name = "loguru" },
{ name = "numpy" },
{ name = "peft" },
{ name = "polars" },
{ name = "tabulate" },
{ name = "torch" },
{ name = "tqdm" },
{ name = "transformers" },
{ name = "tyro" },
{ name = "wandb" },
]
[package.optional-dependencies]
gpu = [
{ name = "vllm" },
]
[package.metadata]
requires-dist = [
{ name = "beartype", specifier = ">=0.18" },
{ name = "causal-conv1d", url = "https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.6.2.post1/causal_conv1d-1.6.2.post1+cu12torch2.8cxx11abiTRUE-cp313-cp313-linux_x86_64.whl" },
{ name = "datasets", specifier = ">=3.0" },
{ name = "einops", specifier = ">=0.8" },
{ name = "flash-attn", url = "https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.7.16/flash_attn-2.8.3%2Bcu128torch2.8-cp313-cp313-linux_x86_64.whl" },
{ name = "flash-linear-attention", specifier = ">=0.5.0" },
{ name = "huggingface-hub", specifier = ">=0.24" },
{ name = "jaxtyping", specifier = ">=0.2" },
{ name = "loguru", specifier = ">=0.7" },
{ name = "numpy", specifier = "<2.0" },
{ name = "peft", specifier = ">=0.13" },
{ name = "polars", specifier = ">=1.0" },
{ name = "tabulate", specifier = ">=0.9" },
{ name = "torch", specifier = ">=2.4" },
{ name = "tqdm", specifier = ">=4.66" },
{ name = "transformers", git = "https://github.com/huggingface/transformers.git?rev=main" },
{ name = "tyro", specifier = ">=0.8" },
{ name = "vllm", marker = "extra == 'gpu'", specifier = ">=0.10" },
{ name = "wandb", specifier = ">=0.18" },
]
provides-extras = ["gpu"]
[[package]]
name = "prometheus-client"
version = "0.25.0"
@@ -2597,7 +2543,7 @@ name = "triton"
version = "3.4.0"
source = { registry = "https://pypi.org/simple" }
dependencies = [
{ name = "setuptools" },
{ name = "setuptools", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')" },
]
wheels = [
{ url = "https://files.pythonhosted.org/packages/30/7b/0a685684ed5322d2af0bddefed7906674f67974aa88b0fae6e82e3b766f6/triton-3.4.0-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:00be2964616f4c619193cb0d1b29a99bd4b001d7dc333816073f92cf2a8ccdeb", size = 155569223, upload-time = "2025-07-30T19:58:44.017Z" },
@@ -2722,6 +2668,60 @@ wheels = [
{ url = "https://files.pythonhosted.org/packages/f8/ba/d69adbe699b768f6b29a5eec7b47dd610bd17a69de51b251126a801369ea/uvloop-0.22.1-cp313-cp313-musllinux_1_2_x86_64.whl", hash = "sha256:1f38ec5e3f18c8a10ded09742f7fb8de0108796eb673f30ce7762ce1b8550cad", size = 4239051, upload-time = "2025-10-16T22:16:43.224Z" },
]
[[package]]
name = "vgrout"
version = "0.1.0"
source = { editable = "." }
dependencies = [
{ name = "beartype" },
{ name = "causal-conv1d" },
{ name = "datasets" },
{ name = "einops" },
{ name = "flash-attn" },
{ name = "flash-linear-attention" },
{ name = "huggingface-hub" },
{ name = "jaxtyping" },
{ name = "loguru" },
{ name = "numpy" },
{ name = "peft" },
{ name = "polars" },
{ name = "tabulate" },
{ name = "torch" },
{ name = "tqdm" },
{ name = "transformers" },
{ name = "tyro" },
{ name = "wandb" },
]
[package.optional-dependencies]
gpu = [
{ name = "vllm" },
]
[package.metadata]
requires-dist = [
{ name = "beartype", specifier = ">=0.18" },
{ name = "causal-conv1d", url = "https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.6.2.post1/causal_conv1d-1.6.2.post1+cu12torch2.8cxx11abiTRUE-cp313-cp313-linux_x86_64.whl" },
{ name = "datasets", specifier = ">=3.0" },
{ name = "einops", specifier = ">=0.8" },
{ name = "flash-attn", url = "https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.7.16/flash_attn-2.8.3%2Bcu128torch2.8-cp313-cp313-linux_x86_64.whl" },
{ name = "flash-linear-attention", specifier = ">=0.5.0" },
{ name = "huggingface-hub", specifier = ">=0.24" },
{ name = "jaxtyping", specifier = ">=0.2" },
{ name = "loguru", specifier = ">=0.7" },
{ name = "numpy", specifier = "<2.0" },
{ name = "peft", specifier = ">=0.13" },
{ name = "polars", specifier = ">=1.0" },
{ name = "tabulate", specifier = ">=0.9" },
{ name = "torch", specifier = ">=2.4" },
{ name = "tqdm", specifier = ">=4.66" },
{ name = "transformers", git = "https://github.com/huggingface/transformers.git?rev=main" },
{ name = "tyro", specifier = ">=0.8" },
{ name = "vllm", marker = "extra == 'gpu'", specifier = ">=0.10" },
{ name = "wandb", specifier = ">=0.18" },
]
provides-extras = ["gpu"]
[[package]]
name = "vllm"
version = "0.11.0"
@@ -2895,8 +2895,8 @@ name = "xformers"
version = "0.0.32.post1"
source = { registry = "https://pypi.org/simple" }
dependencies = [
{ name = "numpy" },
{ name = "torch" },
{ name = "numpy", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')" },
{ name = "torch", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')" },
]
sdist = { url = "https://files.pythonhosted.org/packages/6f/33/3b9c4d3d5b2da453d27de891df4ad653ac5795324961aa3a5c15b0353fe6/xformers-0.0.32.post1.tar.gz", hash = "sha256:1de84a45c497c8d92326986508d81f4b0a8c6be4d3d62a29b8ad6048a6ab51e1", size = 12106196, upload-time = "2025-08-14T18:07:45.486Z" }
wheels = [