张壹 zhangyiss
0 Followers · 0 Following
  • Joined on 2024-09-10
Repositories
54
 Projects Packages Public Activity Starred Repositories
zhangyiss synced commits to refs/pull/1983/head at zhangyiss/mlx from mirror 2025-05-22 13:01:15 +08:00
f7996c25d3 CUDA backend: softmax
de8ae23a36 CUDA backend: layernorm
e346a063ab CUDA backend: indexing ops
171679a176 CUDA backend: sort
35c023e943 CUDA backend: matmul
Compare 13 commits »
zhangyiss synced and deleted reference refs/tags/nit_in_docs at zhangyiss/mlx from mirror 2025-05-22 13:01:14 +08:00
zhangyiss synced and deleted reference refs/tags/refs/pull/2157/merge at zhangyiss/mlx from mirror 2025-05-22 13:01:14 +08:00
zhangyiss synced and deleted reference refs/tags/refs/pull/2214/merge at zhangyiss/mlx from mirror 2025-05-22 13:01:14 +08:00
zhangyiss synced commits to main at zhangyiss/mlx from mirror 2025-05-22 13:01:14 +08:00
55b4062dd8 copyright in docs (#2214)
zhangyiss pushed to dev_yi at zhangyiss/gctl_toolkits 2025-05-22 11:20:58 +08:00
0d7d8311d0 edit Dockerfile
zhangyiss synced commits to nit_in_docs at zhangyiss/mlx from mirror 2025-05-22 04:51:15 +08:00
zhangyiss synced new reference nit_in_docs to zhangyiss/mlx from mirror 2025-05-22 04:51:15 +08:00
zhangyiss synced commits to refs/pull/1983/merge at zhangyiss/mlx from mirror 2025-05-22 04:51:15 +08:00
4da3c5421d Merge 6316472fa4bd3781e8613bf3f8927b6232d3d06d into 79071bfba4
79071bfba4 Fix out-of-bounds default value in logsumexp/softmax (#2213)
7774b87cbd Remove redundant simd_sum in logsumexp (#2210)
Compare 3 commits »
zhangyiss synced and deleted reference refs/tags/refs/pull/2210/merge at zhangyiss/mlx from mirror 2025-05-22 04:51:14 +08:00
zhangyiss synced and deleted reference refs/tags/refs/pull/2213/merge at zhangyiss/mlx from mirror 2025-05-22 04:51:14 +08:00
zhangyiss synced commits to main at zhangyiss/mlx from mirror 2025-05-22 04:51:14 +08:00
79071bfba4 Fix out-of-bounds default value in logsumexp/softmax (#2213)
7774b87cbd Remove redundant simd_sum in logsumexp (#2210)
Compare 2 commits »
zhangyiss synced commits to refs/pull/1983/merge at zhangyiss/mlx from mirror 2025-05-21 20:11:15 +08:00
2a878fcc16 Merge 6316472fa4bd3781e8613bf3f8927b6232d3d06d into 35c87741cf
6316472fa4 Implement logsumexp
823fb8179a Update arg_reduce kernel for https://github.com/ml-explore/mlx/pull/2206
a7f583ffae CUDA backend: softmax
f3810250b2 CUDA backend: layernorm
Compare 39 commits »
zhangyiss synced commits to refs/pull/2074/merge at zhangyiss/mlx from mirror 2025-05-21 20:11:15 +08:00
9f7d576d92 Merge eeaf1fa463 into 35c87741cf
35c87741cf Build for compute capability 70 instead of 75 (#2209)
Compare 2 commits »
zhangyiss synced commits to refs/pull/2156/merge at zhangyiss/mlx from mirror 2025-05-21 20:11:15 +08:00
80976c128c Merge 7c99acb799 into 35c87741cf
35c87741cf Build for compute capability 70 instead of 75 (#2209)
Compare 2 commits »
zhangyiss synced commits to refs/pull/2158/merge at zhangyiss/mlx from mirror 2025-05-21 20:11:15 +08:00
4ec63a560c Merge 3967487b3f0792dfff4e594c60aa73c19bae116d into 35c87741cf
35c87741cf Build for compute capability 70 instead of 75 (#2209)
Compare 2 commits »
zhangyiss synced commits to refs/pull/1983/head at zhangyiss/mlx from mirror 2025-05-21 20:11:14 +08:00
6316472fa4 Implement logsumexp
823fb8179a Update arg_reduce kernel for https://github.com/ml-explore/mlx/pull/2206
a7f583ffae CUDA backend: softmax
f3810250b2 CUDA backend: layernorm
05ede0ea3a CUDA backend: indexing ops
Compare 39 commits »
zhangyiss synced commits to refs/pull/2156/merge at zhangyiss/mlx from mirror 2025-05-21 12:01:16 +08:00
609f3ac010 Merge 7c99acb799 into 4cbe605214
4cbe605214 Feat: Allow per-target Metal debug flags (#2201)
ab8883dd55 include mlx::core::version() symbols in the mlx static library (#2207)
Compare 3 commits »
zhangyiss synced commits to refs/pull/2157/merge at zhangyiss/mlx from mirror 2025-05-21 12:01:16 +08:00
6e23e47758 Merge 4caeb05c64 into 4cbe605214
4cbe605214 Feat: Allow per-target Metal debug flags (#2201)
ab8883dd55 include mlx::core::version() symbols in the mlx static library (#2207)
Compare 3 commits »
zhangyiss synced commits to refs/pull/2158/merge at zhangyiss/mlx from mirror 2025-05-21 12:01:16 +08:00
c18e07174e Merge 3967487b3f0792dfff4e594c60aa73c19bae116d into 4cbe605214
4cbe605214 Feat: Allow per-target Metal debug flags (#2201)
ab8883dd55 include mlx::core::version() symbols in the mlx static library (#2207)
Compare 3 commits »