File size: 280 Bytes
b22e03e | 1 2 3 4 5 6 7 8 | # torch-ext
PyTorch bindings and Python package for `flashrt-flex-attention-train`.
`torch_binding.cpp` registers a package marker so HF `kernel-builder` can load
the extension. The Python module provides the SDPA fallback and the stable
training API used by PI052 integration.
|