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 |
|