kooshi 2bc64597aa model parallelism		2023-03-24 23:03:43 -05:00
..
autograd_4bit.py	model parallelism	2023-03-24 23:03:43 -05:00
gradient_checkpointing.py	Add gradient checkpointing	2023-03-23 08:25:29 +00:00
quant_cuda.cpp	add fast_4bit_matmul and auto switch 2 methods according to bottleneck	2023-03-21 08:43:07 +00:00
quant_cuda_kernel.cu	fix bug	2023-03-23 23:37:39 +08:00