DeepSpeed
Зеркало из https://github.com/microsoft/DeepSpeed
12 строк · 231.0 Байт
1# Copyright (c) Microsoft Corporation.
2# SPDX-License-Identifier: Apache-2.0
3
4# DeepSpeed Team
5
6from .quantize import FP_Quantize, Quantizer
7
8try:
9import triton
10from .fp8_gemm import matmul_fp8
11except ImportError:
12pass
13