bitsandbytes-rocm/bitsandbytes/research/__init__.py

from . import nn
from .autograd._functions import (
    switchback_bnb,
    matmul_fp8_global,
    matmul_fp8_mixed,
)
Refactored simulated fp8 modules into research.nn. 2023-04-12 18:24:44 +00:00			`from . import nn`
Added missing triton and fp8 files. 2023-04-12 17:06:18 +00:00			`from .autograd._functions import (`
			`switchback_bnb,`
			`matmul_fp8_global,`
			`matmul_fp8_mixed,`
			`)`