22 Commits

Author	SHA1	Message	Date
Tim Dettmers	9d353ca786	Merge pull request #87 from lostmsu/main ... Add `device` and `dtype` parameters to `StableEmbedding`	2023-01-02 13:22:45 +01:00
Victor Nova	62d39a237c	add device and dtype parameters to StableEmbedding	2022-11-04 14:12:46 -07:00
Tom Aarsen	54bd6ed1d6	Remove unused imports	2022-10-27 13:32:01 +02:00
Tom Aarsen	0b078403ee	Simplify statements into equivalent, modern variants ... via pyupgrade --py37-plus. The changes e.g. are subclassing from object, calling super() with super(ThisClass, self), or old-style syntax formatting.	2022-10-27 13:14:13 +02:00
Tom Aarsen	1eec77d34c	Remove trailing whitespace & ensure newline at EOF	2022-10-27 13:11:29 +02:00
Tim Dettmers	df86625a93	Isolated CUDASetup logging; all tests green.	2022-10-24 11:54:25 -07:00
justheuristic	5d65817101	debug	2022-09-18 01:09:24 +03:00
dbaranchuk	4dd475ced4	refactoring	2022-09-11 06:28:17 +03:00
dbaranchuk	42b5fc9acc	add memory effcient backward option	2022-09-11 05:51:29 +03:00
dbaranchuk	4d6174bc63	memory efficient fp16 backward	2022-08-25 19:09:23 +03:00
dbaranchuk	ef2936a90d	delete CxB from state	2022-08-24 01:33:04 +03:00
dbaranchuk	656de8ed11	minor fixes	2022-08-23 23:53:43 +03:00
dbaranchuk	8ae9bb23ad	add memory efficient backward	2022-08-23 23:39:54 +03:00
Tim Dettmers	9d60b3c527	Fixed bug in Linear8bitLt, when the bias is None.	2022-08-17 03:45:57 -07:00
Tim Dettmers	de354f7ded	Added fused bias to matmullt.	2022-08-16 12:00:54 -07:00
Titus von Koeller	ea7c14f8ef	reran black with linelength 80 for greater readability	2022-08-01 09:32:47 -07:00
Titus von Koeller	bfa0e33294	ran black and isort for coherent code formatting	2022-08-01 03:31:48 -07:00
Tim Dettmers	c771b3a75a	Most tests passing.	2022-07-22 14:41:05 -07: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
Robin Schmidt	67a1283501	[FIX] passing of sparse in StableEmbedding	2021-11-15 17:27:02 +01:00
Tim Dettmers	7439924891	Initial commit	2021-10-05 19:16:20 -07:00