Tim Dettmers 196d6f5dc1 Merge pull request #469 from shadeMe/linear-layer-device ... Add `device` parameter to `Linear` subclasses and `Embedding`		2023-07-10 06:17:13 -07:00
..
autograd	Extract get_tile_inds to a separate function	2023-06-09 21:39:37 +02:00
cuda_setup	Merge remote-tracking branch 'origin/main' into merge	2023-05-07 13:34:03 -07:00
nn	Merge pull request #469 from shadeMe/linear-layer-device	2023-07-10 06:17:13 -07:00
optim	Added PagedLion and bf16 Lion.	2023-05-23 19:37:38 -07:00
research	Added is_available_triton guard.	2023-04-12 12:10:34 -07:00
triton	Added triton utils.	2023-04-12 12:57:46 -07:00
__init__.py	Merge remote-tracking branch 'origin/main' into merge	2023-05-07 13:34:03 -07:00
__main__.py	Added more detailed cuda setup debug and debugging instructions.	2023-04-11 13:47:10 -07:00
cextension.py	Merge remote-tracking branch 'origin/main' into merge	2023-05-07 13:34:03 -07:00
functional.py	Added PagedLion and bf16 Lion.	2023-05-23 19:37:38 -07:00
utils.py	Merge remote-tracking branch 'origin/main' into merge	2023-05-07 13:34:03 -07:00