Tim Dettmers 55f4c398a0 Polished CUDA SETUP replacement and added docs.		2023-07-14 12:50:59 -07:00
..
autograd	Merge pull request #567 from apbard/patch-1	2023-07-13 21:45:00 -07:00
cuda_setup	Polished CUDA SETUP replacement and added docs.	2023-07-14 12:50:59 -07:00
nn	Added documentation for NF4; failing 8-bit matmul; fixed absmax bug. #529 #543	2023-07-13 21:41:43 -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	Changed CUDA setup to use PyTorch default; added a weak test.	2023-07-13 23:58:41 -07:00
functional.py	Fixed a bug in absmax float conversion.	2023-07-13 21:47:38 -07:00
utils.py	Merge remote-tracking branch 'origin/main' into merge	2023-05-07 13:34:03 -07:00