Chenggang Zhao 78d8362e7a Add a missing #pragma once		2025-05-15 18:10:05 +08:00
..
include/deep_gemm	Add a missing #pragma once	2025-05-15 18:10:05 +08:00
jit	Add __assertfail	2025-05-15 17:04:21 +08:00
jit_kernels	Unify kwargs usages	2025-05-15 16:53:52 +08:00
__init__.py	Weight gradient kernels for dense and MoE models (#95)	2025-05-14 14:47:58 +08:00
utils.py	Weight gradient kernels for dense and MoE models (#95)	2025-05-14 14:47:58 +08:00