k-bit optimizers and matrix multiplication routines.
Project description
bitsandbytes
The bitsandbytes library is a lightweight Python wrapper around CUDA custom functions, in particular 8-bit optimizers, matrix multiplication (LLM.int8()), and 8 & 4-bit quantization functions.
The library includes quantization primitives for 8-bit & 4-bit operations, through bitsandbytes.nn.Linear8bitLt and bitsandbytes.nn.Linear4bit and 8-bit optimizers through bitsandbytes.optim module.
There are ongoing efforts to support further hardware backends, i.e. Intel CPU + GPU, AMD GPU, Apple Silicon. Windows support is quite far along and is on its way as well.
Please head to the official documentation page:
https://huggingface.co/docs/bitsandbytes/main
ALPHA TESTERS WANTED: multi-backend-refactor AMD GPU + Intel CPU/GPU specific BNB backend implementations
We're in the process of a complex refactor in order to allow the support of additional hardware backends, other than CUDA, in BNB. The efforts around this are already quite far along and there's plenty of functionality already in place that is in need for users to take a hands-on approach! Mac support will likely soon also see progress. However, I recommend waiting 2 weeks until the device abstraction has further consolidated (breaking changes upcoming).
Currently, you still need to compile from source, after checking out the multi-backend-refactor branch (instructions WIP, but the current docs on the compilation from source are a good starting point; feel free to share tips / input in this Github discussion. We'll soon enable nightly releases to make this much easier for you!
Please give feedback to us in this dedicated Github Discussion space!
We're super excited about these recent developments and grateful for any constructive input or support that you can give to help us make this a reality. BNB is a community project and we're excited for your collaboration 🤗
License
bitsandbytes is MIT licensed.
We thank Fabio Cannizzo for his work on FastBinarySearch which we use for CPU quantization.
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distributions
File details
Details for the file bitsandbytes-0.43.2-py3-none-win_amd64.whl.
File metadata
- Download URL: bitsandbytes-0.43.2-py3-none-win_amd64.whl
- Upload date:
- Size: 136.5 MB
- Tags: Python 3, Windows x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.8.18
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 89e1c506fd2323574615d668ca7eacad4f83db4847044bf2db87580a71852ff1 |
|
| MD5 | a9697595b65081a5905420ad452069b2 |
|
| BLAKE2b-256 | 867232b06cfe80852d9a60fa8c9169b3712e556ad794d67896022e254d70da56 |
File details
Details for the file bitsandbytes-0.43.2-py3-none-manylinux_2_24_x86_64.whl.
File metadata
- Download URL: bitsandbytes-0.43.2-py3-none-manylinux_2_24_x86_64.whl
- Upload date:
- Size: 137.5 MB
- Tags: Python 3, manylinux: glibc 2.24+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.8.18
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 80fbc0f41dded735f51328042d3d45ea640101d87c8abba8ea5bfafa61e2a786 |
|
| MD5 | 58ab4f2e44365e58eb31cf2efe046fcb |
|
| BLAKE2b-256 | 72e506ed351cdf8d1d5bf7eb86729e4e3669c6844654354aef3b1bc9da66d0bb |
