张壹 zhangyiss
0 Followers · 0 Following
  • Joined on 2024-09-10
Repositories
54
 Projects Packages Public Activity Starred Repositories
zhangyiss synced commits to main at zhangyiss/mlx from mirror 2025-07-06 01:11:13 +08:00
0e0d9ac522 [CUDA] Add MLX_CUDA_GRAPH_CACHE_SIZE env for setting graph cache size (#2329)
zhangyiss synced commits to moe-backward at zhangyiss/mlx from mirror 2025-07-06 01:11:13 +08:00
9e5bb5295a Add more tests and fix qmm gradient
zhangyiss synced commits to refs/pull/1970/merge at zhangyiss/mlx from mirror 2025-07-06 01:11:13 +08:00
9fe9ff6f15 Merge 688e421184 into 0e0d9ac522
0e0d9ac522 [CUDA] Add MLX_CUDA_GRAPH_CACHE_SIZE env for setting graph cache size (#2329)
8917022deb fix graphs for older cuda (#2328)
ec0d5db67b [CUDA] Switch to CUDA graphs (#2317)
e76e9b87f0 Fix compilation error from integral_constant (#2326)
Compare 9 commits »
zhangyiss synced and deleted reference refs/tags/refs/pull/2329/merge at zhangyiss/mlx from mirror 2025-07-06 01:11:12 +08:00
zhangyiss synced commits to moe-backward at zhangyiss/mlx from mirror 2025-07-05 16:51:14 +08:00
3d4174cd37 Add gradient for the scales and biases in gather qmm
bda1534a44 Improve the gradient of gather_qmm as well
b28577289e Disable the test for CUDA
2d0f452aae Fix the test and cpu edge case
Compare 4 commits »
zhangyiss synced commits to refs/pull/2329/head at zhangyiss/mlx from mirror 2025-07-05 16:51:14 +08:00
08b4bffbed [CUDA] Add MLX_CUDA_GRAPH_CACHE_SIZE env for setting graph cache size
zhangyiss synced commits to refs/pull/2329/merge at zhangyiss/mlx from mirror 2025-07-05 16:51:14 +08:00
66e8e67f05 Merge 08b4bffbed into 8917022deb
08b4bffbed [CUDA] Add MLX_CUDA_GRAPH_CACHE_SIZE env for setting graph cache size
Compare 2 commits »
zhangyiss synced commits to moe-backward at zhangyiss/mlx from mirror 2025-07-05 08:41:13 +08:00
bd0622c4d9 Address floating point exception on linux blas
zhangyiss synced commits to refs/pull/2074/merge at zhangyiss/mlx from mirror 2025-07-05 00:31:15 +08:00
0045d8c17c Merge eeaf1fa463 into 8917022deb
8917022deb fix graphs for older cuda (#2328)
ec0d5db67b [CUDA] Switch to CUDA graphs (#2317)
e76e9b87f0 Fix compilation error from integral_constant (#2326)
cfb6a244ea allow parameters to be deleted (#2325)
Compare 5 commits »
zhangyiss synced commits to gh-pages at zhangyiss/YouPlot from mirror 2025-07-04 17:41:14 +08:00
b2be62e002 deploy: 1c80af2b2f
de8addf117 deploy: 2d8cbba4c2
8e3aa799f8 deploy: 8a8a009b22
Compare 3 commits »
zhangyiss synced commits to main at zhangyiss/YouPlot from mirror 2025-07-04 17:41:14 +08:00
1c80af2b2f Update sampo grafiikka URL
2d8cbba4c2 Update LICENSE.txt
8a8a009b22 Removed FUNDING.yml
Compare 3 commits »
zhangyiss synced commits to moe-backward at zhangyiss/mlx from mirror 2025-07-04 16:21:14 +08:00
22f9b8a6fc More JIT fixes
zhangyiss synced commits to refs/pull/2329/head at zhangyiss/mlx from mirror 2025-07-04 16:21:14 +08:00
d8f4afe935 Disable graph cache for CUDA < 12.4
zhangyiss synced commits to refs/pull/2329/merge at zhangyiss/mlx from mirror 2025-07-04 16:21:14 +08:00
9d52b32a12 Merge d8f4afe93557abea68365bf83c7f0a1d3aa335eb into 8917022deb
d8f4afe935 Disable graph cache for CUDA < 12.4
Compare 2 commits »
zhangyiss synced commits to moe-backward at zhangyiss/mlx from mirror 2025-07-04 08:11:13 +08:00
aaf44d915f Add no cpu and no gpu implementations where needed
ed1a81210d Fix the JIT
f2994d5b29 Fix scatter_axis on single axis arrays
06a2e74eb2 Fix metal jit
d96a33c776 Add rudimentary test for gather_mm with sorted indices
Compare 7 commits »
zhangyiss synced commits to refs/pull/2104/merge at zhangyiss/mlx from mirror 2025-07-04 08:11:13 +08:00
7440ca7e89 Merge 992eac905a into 8917022deb
8917022deb fix graphs for older cuda (#2328)
Compare 2 commits »
zhangyiss synced commits to refs/pull/900/merge at zhangyiss/FTXUI from mirror 2025-07-04 07:01:13 +08:00
1b4248db4f Merge baa5973128 into b78b97056b
b78b97056b Stop using Sender/Receiver in TerminalInputParser. (#1073)
Compare 2 commits »
zhangyiss synced commits to refs/pull/2297/merge at zhangyiss/mlx from mirror 2025-07-03 15:51:15 +08:00
e0a4db504d Merge 043c37cccd into 8917022deb
8917022deb fix graphs for older cuda (#2328)
ec0d5db67b [CUDA] Switch to CUDA graphs (#2317)
e76e9b87f0 Fix compilation error from integral_constant (#2326)
Compare 4 commits »
zhangyiss synced commits to refs/pull/2300/merge at zhangyiss/mlx from mirror 2025-07-03 15:51:15 +08:00
47abf946c4 Merge cc4de6a607 into 8917022deb
8917022deb fix graphs for older cuda (#2328)
ec0d5db67b [CUDA] Switch to CUDA graphs (#2317)
e76e9b87f0 Fix compilation error from integral_constant (#2326)
Compare 4 commits »
zhangyiss synced commits to refs/pull/2328/head at zhangyiss/mlx from mirror 2025-07-03 15:51:15 +08:00
136f0792fa fix graphs for older cuda