| # Copyright (c) Facebook, Inc. and its affiliates. | |
| # | |
| # This source code is licensed under the MIT license found in the | |
| # LICENSE file in the root directory of this source tree. | |
| from . import research, utils | |
| from .autograd._functions import ( | |
| MatmulLtState, | |
| bmm_cublas, | |
| matmul, | |
| matmul_4bit, | |
| matmul_cublas, | |
| mm_cublas, | |
| ) | |
| from .nn import modules | |
| from .optim import adam | |
| __pdoc__ = { | |
| "libbitsandbytes": False, | |
| "optim.optimizer.Optimizer8bit": False, | |
| "optim.optimizer.MockArgs": False, | |
| } | |
| __version__ = "0.43.1" | |