extensions/pybind/softmax - hpcaitech/ColossalAI

Making large AI models cheaper, faster and more accessible

41377 0 1 Python

	__init__.py	271 B
	scaled_masked_softmax_cuda.py	1002 B
	scaled_masked_softmax.cpp	2.0 KB
	scaled_upper_triang_masked_softmax.cpp	1.7 KB
	scaled_upper_triangle_masked_softmax_cuda.py	1.0 KB