Q-TensorFormer / kernels /__init__.py
Premchandyadav369
feat: synchronize space repo with 6 frontier research modules
7975a75
Raw History Blame Contribute Delete
337 Bytes
"""
Kernel acceleration module for Q-TensorFormer.
"""
from .tt_kernels import (
slice_tt_cores,
optimized_tt_contract,
fast_tt_linear,
benchmark_tt_kernel,
TRITON_AVAILABLE,
)
__all__ = [
"slice_tt_cores",
"optimized_tt_contract",
"fast_tt_linear",
"benchmark_tt_kernel",
"TRITON_AVAILABLE",
]