8 Commits
Author SHA1 Message Date
JinjieLiu c0eba77ed6 switch backend to setuptools
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
2026-04-20 10:23:36 +08:00
JinjieLiu a727526794 add length check to unwrap arguments
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
2026-02-12 15:31:50 +08:00
JinjieLiu 599957e156 support attention bwd
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
2026-02-10 17:01:28 +08:00
JinjieLiu e41ec26329 support attention fwd
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
2026-02-09 17:15:51 +08:00
JinjieLiu f6c7a48c1b enable lambda function for grid descriptor
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
2026-02-05 15:59:22 +08:00
JinjieLiu b7bf598fde support softmax
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
2026-02-04 16:38:33 +08:00
JinjieLiu 192dc95ac0 supports decorator for jit and wrapper
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
2026-02-04 10:46:14 +08:00
JinjieLiu 6e4c2d4a43 fix bugs on lacking cudatoolkit
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
2026-02-04 10:27:44 +08:00