Phil Wang c83888aa1a use epsilon as beta2 for lion, complete most of the logic in kernel.cu for all functions		2023-03-09 11:54:54 -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	use epsilon as beta2 for lion, complete most of the logic in kernel.cu for all functions	2023-03-09 11:54:54 -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