125 Commits

Author	SHA1	Message	Date
Tim Dettmers	dd50382b32	Full evaluate_cuda setup with integration test.	2022-07-31 17:47:44 -07:00
Titus von Koeller	5d90b38c4d	adding CLI tool for CUDA install debugging - intermediate commit	2022-07-27 21:16:04 -07:00
Tim Dettmers	389f66ca5a	Fixed direct extraction masking.	2022-07-27 01:46:35 -07:00
Tim Dettmers	5737f2b027	Merge branch 'patch_merge' into extract_outliers	2022-07-26 19:38:01 -07:00
Tim Dettmers	47a73d94c3	Matmullt with direct outlier extraction for 8-bit inference.	2022-07-26 19:15:35 -07:00
Tim Dettmers	cbb901ac51	Boilerplate and test for extract_outliers.	2022-07-26 12:12:38 -07:00
Tim Dettmers	1e88edd8c0	Removed rowscale (segfaults on ampere).	2022-07-25 17:27:57 -07:00
Tim Dettmers	c771b3a75a	Most tests passing.	2022-07-22 14:41:05 -07:00
Max Ryabinin	e4cf33f2a3	Fix imports	2022-07-01 17:25:44 +03:00
Max Ryabinin	8258b4364a	Add a CPU-only build option	2022-07-01 17:16:10 +03:00
Max Ryabinin	33efe4a09f	Remove unused imports, fix NotImplementedError	2022-06-30 18:14:20 +03:00
Tim Dettmers	20e1677dfd	Added module override, bnb.nn.Embedding #13 #15 #19	2021-11-29 09:32:13 -08:00
Tim Dettmers	3cff6795fb	Merge branch 'main' of github.com:facebookresearch/bitsandbytes into 0.26.0	2021-11-29 08:24:17 -08:00
Tim Dettmers	108cf9fc1f	Fixed unsafe use of eval. #8	2021-11-29 08:21:05 -08:00
Tim Dettmers	2f8083bd8b	Added AdamW. #10 #13	2021-11-28 21:18:11 -08:00
Robin Schmidt	67a1283501	[FIX] passing of sparse in StableEmbedding	2021-11-15 17:27:02 +01:00
Tim Dettmers	8b3c0f355c	Added adagrad with tests (no clipping).	2021-11-10 15:10:02 -08:00
Konstantin Ditschuneit	56f5274848	Add missing imports to adam	2021-10-31 16:38:38 +01:00
Tim Dettmers	0c5fa5a637	Fixed syntax and import error.	2021-10-21 16:04:18 -07:00
Tim Dettmers	1ec0d54529	Added analysis Adam.	2021-10-21 15:53:20 -07:00
Tim Dettmers	eaf35ab949	Copied over Analysis Adam.	2021-10-21 10:20:41 -07:00
Tim Dettmers	a6eae2e7f2	Added skip_zeros; tests are passing.	2021-10-20 19:15:47 -07:00
Tim Dettmers	bb34fd50a1	Initial plumbing for skip_zeros.	2021-10-20 18:37:44 -07:00
Tim Dettmers	7923c4a066	Changed from testpypi to pypi. Release 0.0.24	2021-10-07 08:39:38 -07:00
Tim Dettmers	7439924891	Initial commit	2021-10-05 19:16:20 -07:00