Phil Wang c99b44f774 do the epsilon beta2 switcharoo within the cuda code, and not within the python class (so that the state dict still makes sense)		2023-03-10 08:57:59 -08:00
..
autograd	Added Int8 matmul support for all GPUs. Full backward support.	2023-02-01 20:09:31 -08:00
cuda_setup	Added version 0.37.0.	2023-02-01 20:27:01 -08:00
nn	Added Int8 matmul support for all GPUs. Full backward support.	2023-02-01 20:09:31 -08:00
optim	do the epsilon beta2 switcharoo within the cuda code, and not within the python class (so that the state dict still makes sense)	2023-03-10 08:57:59 -08:00
__init__.py	Sort imports	2022-10-27 13:15:21 +02:00
__main__.py	Sort imports	2022-10-27 13:15:21 +02:00
cextension.py	Fixed issue where the CUDA SETUP was not printed.	2023-01-04 03:50:53 -08:00
functional.py	do a bunch of typical bookkeeping before getting to main lion logic	2023-03-09 10:10:19 -08:00
utils.py	Replace print_stderr with warnings.warn	2022-08-24 18:45:17 +03:00