10 Commits

Author	SHA1	Message	Date
Tim Dettmers	2f01865a2f	Added CUDA block assert and is_on_gpu check.	2022-08-03 09:05:37 -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	8b3c0f355c	Added adagrad with tests (no clipping).	2021-11-10 15:10:02 -08: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	7439924891	Initial commit	2021-10-05 19:16:20 -07:00